Fixed cifar10

Author: moelife-coder

Train.py

@@ -24,9 +24,9 @@ def train(EPOCH, dataloader, optimizer, loss_function, network, model_name, nega
             print('Epoch: ', epoch + 1, '| Step: ', step + 1, '| Train loss: %.4f' % loss.cpu().data.numpy())
             if step % SAVE_STEP == 1:
                 print('Saving models...')
-                torch.save(network, model_name)
+                torch.save(network, "{}-{}-step{}.pkl".format(model_name, epoch, step))
         print('Saving models...')
-        torch.save(network, model_name)
+        torch.save(network, "{}-{}.pkl".format(model_name, epoch))
         print('Saving logs...')
         f.write("{} {}\n".format(epoch, loss.cpu().data.numpy()))
 

run.sh

@@ -9,8 +9,8 @@ train_cnn_mnist() {
   python Train.py --epoch 20 --dataset mnist --network cnn --loss_func "$2" --model_name "$3" 1 "$1"
 }
 train_mlp_cifar() {
-  python Train.py --epoch 20 --dataset cifar10 --network mlp --loss_func "$2" --model_name "$3" 784 "$1"
+  python Train.py --epoch 20 --dataset cifar10 --network mlp --loss_func "$2" --model_name "$3" 3072 "$1"
 }
 train_cnn_cifar() {
-  python Train.py --epoch 20 --dataset cifar10 --network cnn --loss_func "$2" --model_name "$3" 1 "$1"
+  python Train.py --epoch 20 --dataset cifar10 --network cnn --loss_func "$2" --model_name "$3" 3 "$1"
 }