Skip to content

Commit f276dd3

Browse files
committed
fixed the num epochs for imagenet
1 parent 0169e2c commit f276dd3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cnn/train_imagenet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
parser.add_argument('--weight_decay', type=float, default=3e-5, help='weight decay')
2828
parser.add_argument('--report_freq', type=float, default=100, help='report frequency')
2929
parser.add_argument('--gpu', type=int, default=0, help='gpu device id')
30-
parser.add_argument('--epochs', type=int, default=1000, help='num of training epochs')
30+
parser.add_argument('--epochs', type=int, default=250, help='num of training epochs')
3131
parser.add_argument('--init_channels', type=int, default=48, help='num of init channels')
3232
parser.add_argument('--layers', type=int, default=14, help='total number of layers')
3333
parser.add_argument('--auxiliary', action='store_true', default=False, help='use auxiliary tower')

0 commit comments

Comments
 (0)