changing default params

Author: jordanlei

train_coco.py

@@ -13,7 +13,7 @@
 parser.add_argument('--epochs', type = int, default=100)
 parser.add_argument('--randomseed', type = int, default=100)
 parser.add_argument('--lr', type = float, default=0.0001)
-parser.add_argument('--penalty', type = float, default=10000)
+parser.add_argument('--penalty', type = float, default=5000)
 
 args = parser.parse_args()
 run_id = args.name

train_mnist.py

@@ -8,7 +8,7 @@
 parser.add_argument('--strength', type= float, default = 0.2)
 parser.add_argument('--noise', type= float, default = 0.3)
 parser.add_argument('--name', type = str, default='mnist_model')
-parser.add_argument('--randomseed', type = int, default=2021)
+parser.add_argument('--randomseed', type = int, default=100)
 parser.add_argument('--epochs', type = int, default=10)
 
 args = parser.parse_args()
@@ -19,7 +19,7 @@
 f.write("\n")
 f.close()
 
-# torch.manual_seed(args.randomseed)
+torch.manual_seed(args.randomseed)
 
 def printwrite(x): 
     print(x)