diff --git a/tools/train.py b/tools/train.py
index 3ad78e1b4159b09f87cd45b7a86f2e91210d9500..f4c00714c259f69d6592b156b67a55f9879c7b06 100644
--- a/tools/train.py
+++ b/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')