Abdullah-Nazhat
/

Normalizer

Model card Files Files and versions

Abdullah-Nazhat commited on Aug 24

Commit

a748455

·

verified ·

1 Parent(s): 0d02665

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -145,4 +145,4 @@ print("Done!")
 path = "/PATH/Normalizer/Experiments_cifar10/weights_normalizer"
 model_name = "NormalizerImageClassification_cifar10"
 torch.save(model.state_dict(), f"{path}/{model_name}.pth")
-print(f"Saved Model State to {path}/{model_name}.pth ")

 path = "/PATH/Normalizer/Experiments_cifar10/weights_normalizer"
 model_name = "NormalizerImageClassification_cifar10"
 torch.save(model.state_dict(), f"{path}/{model_name}.pth")
+print(f"Saved Model State to {path}/{model_name}.pth ")