Update main.py

lshiwjx · Apr 9, 2018 · d3995ce · d3995ce
1 parent 74672f1
commit d3995ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -356,7 +356,7 @@ def eval(self, epoch, save_score=False, loader_name=['test']):
 
             if save_score:
                 with open('{}/epoch{}_{}_score.pkl'.format(
-                        self.arg.work_dir, epoch + 1, ln), 'w') as f:
+                        self.arg.work_dir, epoch + 1, ln), 'wb') as f:
                     pickle.dump(score_dict, f)
 
     def start(self):