thebears/inaturalist_pytorch_model
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

yacwc

Browse Source
This commit is contained in:
Ishan S. Patel
2021-07-12 13:00:21 -04:00
parent f46d193826
commit 3e54faeb44
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
data.py
View File

@@ -15,7 +15,8 @@ device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cp
if sys.platform == "win32":
PATH_ROOT = r"D:\ishan\ml\inaturalist\\"
else:
raise NotImplementedError("Not defined for this platform")
PATH_ROOT = '/home/thebears/data/ml/inaturalist'
def get_transform(train):