raja5259 commited on
Commit
cb5b79f
·
verified ·
1 Parent(s): 8af2a3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "weights.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
 
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