fix

kamiQiu · Jun 1, 2018 · 88e4148 · 88e4148
1 parent ee2a247
commit 88e4148
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,5 @@
 #custom
+tmp*
 work_dir/*
 data/*
 config_v0/*

diff --git a/main.py b/main.py
@@ -141,12 +141,12 @@ def get_parser():
     parser.add_argument(
         '--display_by_category',
         type=str2bool,
-        default=True,
+        default=False,
         help='if ture, the top k accuracy by category  will be displayed')
     parser.add_argument(
         '--display_recall_precision',
         type=str2bool,
-        default=True,
+        default=False,
         help='if ture, recall and precision by category  will be displayed')
 
     return parser
@@ -341,7 +341,6 @@ def eval(self, epoch, save_score=False, loader_name=['test']):
             loss_value = []
             score_frag = []
             for batch_idx, (data, label) in enumerate(self.data_loader[ln]):
-                print('{}/{}'.format(batch_idx, len(self.data_loader[ln])))
                 data = Variable(
                     data.float().cuda(self.output_device),
                     requires_grad=False,