Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ from torchvision import datasets, transforms, utils
|
|
35 |
|
36 |
save1_or_load0 = False
|
37 |
|
38 |
-
model, trainer = ts_lt(save1_or_load0, Epochs = 26, wt_fname = "
|
39 |
'''
|
40 |
ts_lt(save1_or_load0, # decision maker for training Vs testing
|
41 |
Epochs = 1, # argument for training
|
|
|
35 |
|
36 |
save1_or_load0 = False
|
37 |
|
38 |
+
model, trainer = ts_lt(save1_or_load0, Epochs = 26, wt_fname = "weights_92.ckpt") # Train and Save Vs Load and Test
|
39 |
'''
|
40 |
ts_lt(save1_or_load0, # decision maker for training Vs testing
|
41 |
Epochs = 1, # argument for training
|