From 3e54faeb44051bf53c4dd48171ddbb0148d80556 Mon Sep 17 00:00:00 2001
From: "Ishan S. Patel" <ispatel@live.com>
Date: Mon, 12 Jul 2021 13:00:21 -0400
Subject: [PATCH] yacwc

---
 data.py  | 3 ++-
 train.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/data.py b/data.py
index e71666a..b9bb93f 100644
--- a/data.py
+++ b/data.py
@@ -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):
diff --git a/train.py b/train.py
index aa0d344..9207eb4 100644
--- a/train.py
+++ b/train.py
@@ -45,7 +45,7 @@ def run():
 
     train_data_loader = torch.utils.data.DataLoader(
         train_dataset,
-        batch_size=8,
+        batch_size=16,
         shuffle=True,
         num_workers=4,
         collate_fn=utils.collate_fn,
@@ -53,7 +53,7 @@ def run():
 
     val_data_loader = torch.utils.data.DataLoader(
         val_dataset,
-        batch_size=8,
+        batch_size=16,
         shuffle=True,
         num_workers=4,
         collate_fn=utils.collate_fn,