change main

Code/main.py

@@ -67,7 +67,7 @@ def run(raw_args):
 	#train...
 	print("training..")
 	if args.train_loop=="swp":
-		evaluation_test, evaluation_train = train.train(model, args.architecture,args.random_seed, args.gradient_accumulation_steps, args.mix_up, args.threshold, args.lambda_value, args.mixup_epoch, args.tmix, args.mixlayer, train_dataset, test_dataset, args.epochs, args.learning_rate, args.batch_size, args.test_batch_size)
+		evaluation_test, evaluation_train = train.train(model, args.architecture,args.random_seed, args.gradient_accumulation_steps, args.mix_up, args.threshold, args.lambda_value, args.mixepoch, args.tmix, args.mixlayer, train_dataset, test_dataset, args.epochs, args.learning_rate, args.batch_size, args.test_batch_size)
 	elif args.train_loop=="salami":
 		evaluation_test = train.train_salami(model,args.random_seed, train_dataset, test_dataset, args.batch_size, args.test_batch_size, args.learning_rate, args.epochs)
 	else:
@@ -141,7 +141,6 @@ if __name__ == "__main__":
 		help="whether or not to preprocess the test set with context")
 
 	parser.add_argument(
-		"-m",
 		"--masking",
 		#default=False,
 		#type=bool,
@@ -214,8 +213,8 @@ if __name__ == "__main__":
 		default=0.4)
 
 	parser.add_argument(
-		"-mixup_epoch", 
-		"--mixup_epoch", 
+		"-mixepoch", 
+		"--mixepoch", 
 		help="specify the epoch(s) in which to apply mixup",
 		type=int,
 		default=1)