--- a +++ b/settings.py @@ -0,0 +1,7 @@ +batch_size = 64 +epochs = 50 +workers = 4 +lr = 0.001 +momentum=0.9 +step_size=7 +gamma=0.1