From 73131af703a67f0d8e976e4cc03b40d52224d350 Mon Sep 17 00:00:00 2001 From: Max deGroot Date: Wed, 29 Mar 2017 18:35:35 -0500 Subject: [PATCH] reset defaults --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 2cc2b8d7f..8c68da2e7 100644 --- a/train.py +++ b/train.py @@ -21,7 +21,7 @@ parser.add_argument('--version', default='v2', help='conv11_2(v2) or pool6(v1) as last layer') parser.add_argument('--basenet', default='vgg16_reducedfc.pth', help='pretrained base model') parser.add_argument('--jaccard_threshold', default=0.5, type=float, help='Min Jaccard index for matching') -parser.add_argument('--batch_size', default=16, type=int, help='Batch size for training') +parser.add_argument('--batch_size', default=32, type=int, help='Batch size for training') parser.add_argument('--num_workers', default=4, type=int, help='Number of workers used in dataloading') parser.add_argument('--iterations', default=120000, type=int, help='Number of training epochs') parser.add_argument('--cuda', default=True, type=bool, help='Use cuda to train model') @@ -30,7 +30,7 @@ parser.add_argument('--weight_decay', default=5e-4, type=float, help='Weight decay for SGD') parser.add_argument('--gamma', default=0.1, type=float, help='Gamma update for SGD') parser.add_argument('--log_iters', default=True, type=bool, help='Print the loss at each iteration') -parser.add_argument('--visdom', default=True, type=bool, help='Use visdom to for loss visualization') +parser.add_argument('--visdom', default=False, type=bool, help='Use visdom to for loss visualization') parser.add_argument('--save_folder', default='weights/', help='Location to save checkpoint models') args = parser.parse_args()