diff --git a/main.py b/main.py index c8c7c42..03c96ec 100644 --- 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')