blblbl

filippo-merlo · Feb 6, 2024 · b52c4f2 · b52c4f2
1 parent 88418e7
commit b52c4f2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -279,9 +279,9 @@ def my_clip_train(in_path, out_path, model_name, source, in_base,
 
 	# Train without 9 combs and 3 attrs
 	model_name = 'comparative_base_mem.pickle'
-	#my_clip_train(args.in_path, args.out_path, model_name,
-	#			'novel_train/', bn_n_train, ['rgba'], dic_train, vocabs, args.pre_train)
-	#
+	my_clip_train(args.in_path, args.out_path, model_name,
+				'novel_train/', bn_n_train, ['rgba'], dic_train, vocabs, args.pre_train)
+
 	# Multi-Attribute Recognition Evaluation
 	in_memory = os.path.join(args.out_path, model_name)
 	infile = open(in_memory, 'rb')