Update code.

soujanyaporia · Sep 5, 2018 · 22e7e30 · 22e7e30
1 parent bbf3546
commit 22e7e30
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/run.py b/run.py
@@ -392,10 +392,11 @@ def str2bool(v):
         with open('unimodal_{0}_{1}way.pickle'.format(args.data, args.classes), 'wb') as handle:
             pickle.dump(unimodal_activations, handle, protocol=pickle.HIGHEST_PROTOCOL)
 
-    with open('unimodal_{0}_{1}way.pickle'.format(args.data, args.classes), 'rb') as handle:
-        u = pickle._Unpickler(handle)
-        u.encoding = 'latin1'
-        unimodal_activations = u.load()
+    if not args.use_raw:
+        with open('unimodal_{0}_{1}way.pickle'.format(args.data, args.classes), 'rb') as handle:
+            u = pickle._Unpickler(handle)
+            u.encoding = 'latin1'
+            unimodal_activations = u.load()
 
     epochs = 50
     multimodal(unimodal_activations, args.data, args.classes, args.fusion, args.attention_2, use_raw=args.use_raw)