Abdullah-Nazhat commited on
Commit
6739459
1 Parent(s): 22b5fbe

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +2 -2
train.py CHANGED
@@ -160,7 +160,7 @@ def test(dataloader, model, loss_fn):
160
 
161
 
162
 
163
- logname = "/home/abdullah/Desktop/Normalizer/Experiments_cifar10/logs_normalizer/logs_cifar10.csv"
164
  if not os.path.exists(logname):
165
  with open(logname, 'w') as logfile:
166
  logwriter = csv.writer(logfile, delimiter=',')
@@ -181,7 +181,7 @@ print("Done!")
181
 
182
 
183
 
184
- path = "/home/abdullah/Desktop/Normalizer/Experiments_cifar10/weights_normalizer"
185
  model_name = "NormalizerImageClassification_cifar10"
186
  torch.save(model.state_dict(), f"{path}/{model_name}.pth")
187
  print(f"Saved Model State to {path}/{model_name}.pth ")
 
160
 
161
 
162
 
163
+ logname = "/PATH/Normalizer/Experiments_cifar10/logs_normalizer/logs_cifar10.csv"
164
  if not os.path.exists(logname):
165
  with open(logname, 'w') as logfile:
166
  logwriter = csv.writer(logfile, delimiter=',')
 
181
 
182
 
183
 
184
+ path = "/PATH/Normalizer/Experiments_cifar10/weights_normalizer"
185
  model_name = "NormalizerImageClassification_cifar10"
186
  torch.save(model.state_dict(), f"{path}/{model_name}.pth")
187
  print(f"Saved Model State to {path}/{model_name}.pth ")