Skip to content

Commit 498f297

Browse files
authored
Update train_test.py
1 parent 8a3d82f commit 498f297

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,8 @@ def str2bool(v):
4545
default=4e-3, type=float, help='initial learning rate')
4646
parser.add_argument('--momentum', default=0.9, type=float, help='momentum')
4747

48-
parser.add_argument('--resume_net', default=True, help='resume net for retraining')
49-
parser.add_argument('--resume_epoch', default=10,
48+
parser.add_argument('--resume_net', default=False, help='resume net for retraining')
49+
parser.add_argument('--resume_epoch', default=0,
5050
type=int, help='resume iter for retraining')
5151

5252
parser.add_argument('-max','--max_epoch', default=300,

0 commit comments

Comments
 (0)