hjb

filippo-merlo · Feb 6, 2024 · 27eb5a0 · 27eb5a0
1 parent c8fcc8c
commit 27eb5a0
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')