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": if sys.platform == "win32":
PATH_ROOT = r"D:\ishan\ml\inaturalist\\" PATH_ROOT = r"D:\ishan\ml\inaturalist\\"
else: else:
raise NotImplementedError("Not defined for this platform") PATH_ROOT = '/home/thebears/data/ml/inaturalist'
def get_transform(train): def get_transform(train):

4
train.py
View File

@@ -45,7 +45,7 @@ def run():
train_data_loader = torch.utils.data.DataLoader( train_data_loader = torch.utils.data.DataLoader(
train_dataset, train_dataset,
batch_size=8, batch_size=16,
shuffle=True, shuffle=True,
num_workers=4, num_workers=4,
collate_fn=utils.collate_fn, collate_fn=utils.collate_fn,
@@ -53,7 +53,7 @@ def run():
val_data_loader = torch.utils.data.DataLoader( val_data_loader = torch.utils.data.DataLoader(
val_dataset, val_dataset,
batch_size=8, batch_size=16,
shuffle=True, shuffle=True,
num_workers=4, num_workers=4,
collate_fn=utils.collate_fn, collate_fn=utils.collate_fn,