n epoch

tools/train.py

@@ -40,7 +40,7 @@ parser.add_argument('--decay_margin', default=0.016, help='margin to decay lr &
 parser.add_argument('--refine_margin', default=0.013, help='margin to start the training of iterative refinement')
 parser.add_argument('--noise_trans', default=0.03, help='range of the random noise of translation added to the training data')
 parser.add_argument('--iteration', type=int, default = 2, help='number of refinement iterations')
-parser.add_argument('--nepoch', type=int, default=60, help='max number of epochs to train')
+parser.add_argument('--nepoch', type=int, default=200, help='max number of epochs to train')
 parser.add_argument('--resume_posenet', type=str, default = '',  help='resume PoseNet model')
 parser.add_argument('--resume_refinenet', type=str, default = '',  help='resume PoseRefineNet model')
 parser.add_argument('--start_epoch', type=int, default = 1, help='which epoch to start')
@@ -62,8 +62,8 @@ def main():
         opt.num_objects = 8 #TODO
         #opt.num_objects = 5 #TODO
         opt.num_points = 500
-        opt.outf = 'trained_models/linemod8'
-        opt.log_dir = 'experiments/logs/linemod8'
+        opt.outf = 'trained_models/linemod'
+        opt.log_dir = 'experiments/logs/linemod'
         opt.repeat_epoch = 5
     else:
         print('Unknown dataset')