Update config.py
Browse files
config.py
CHANGED
@@ -9,7 +9,7 @@ EPOCHS = 5
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
-
MODEL_PATH = "
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|
|
|
9 |
|
10 |
# Folder to contain all the datasets
|
11 |
DATASET_LOCATION = "" #
|
12 |
+
MODEL_PATH = "model.bin"
|
13 |
|
14 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
15 |
|