Commit Graph

10 Commits

Author SHA1 Message Date
Xuebin Qin
7bf8a3be2d
Merge pull request #6 from deshwalmahesh/main
gt.cuda() creating error for CPU. Change to .to(device)
2022-07-30 09:27:04 -07:00
PiggyJerry
db68ff3480 update data_loader_cache.py 2022-07-27 23:54:20 +04:00
PiggyJerry
6e807f8d10 update data_loader_cache.py 2022-07-27 23:42:02 +04:00
Xuebin Qin
362f4e9699
Update data_loader_cache.py
temporarily solution for training bugs
2022-07-26 19:55:35 -07:00
Xuebin Qin
85a9a1be60 dis v2 2022-07-17 16:19:20 -07:00
Xuebin Qin
b9e31bdc26 dis v2 2022-07-17 16:14:28 -07:00
Xuebin Qin
7c61e1876d dataloader for inference w/o gt 2022-07-17 13:17:27 -07:00
Xuebin Qin
c1480958ac compute_loss 2022-07-17 11:00:12 -07:00
Mahesh Deshwal
8836d5b4f4 gt.cuda() creating error for CPU. Change to .to(device) 2022-07-17 14:52:08 +05:30
Xuebin Qin
e6455bae38 official release of our isnet and dis5k 2022-07-16 22:56:37 -07:00