From 4852ee5043a46d59f7b4337426e0d14cbdde225e Mon Sep 17 00:00:00 2001
From: Guillaume-Duret <guillaume.duret@ec-lyon.fr>
Date: Wed, 14 Jun 2023 10:25:37 +0200
Subject: [PATCH] n epoch
---
tools/train.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tools/train.py b/tools/train.py
index 3ad78e1..f4c0071 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')
--
GitLab