tarin loops

Code/main.py

@@ -55,9 +55,12 @@ def run(raw_args):
 		print("non eligible tokenizer selected")
 
 	#train...
-	evaluation_test, evaluation_train = train.train(model, args.random_seed, train_dataset, test_dataset, args.epochs, args.learning_rate, args.batch_size, args.test_batch_size)
-	evaluation_test = train_salami(model, train_set, test_set, args.batch_size, args.test_batch_size, args.learning_rate, args.epochs)
-
+	if args.train_loop=="swp":
+		evaluation_test, evaluation_train = train.train(model, args.random_seed, train_dataset, test_dataset, args.epochs, args.learning_rate, args.batch_size, args.test_batch_size)
+	elif args.train_loop=="salami":
+		evaluation_test = train_salami(model, train_set, test_set, args.batch_size, args.test_batch_size, args.learning_rate, args.epochs)
+	else:
+		print("no eligible train loop selected")
 	#(evaluate... is done internally) but could maybe be implemented here to make average over multiple random seeds