Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-07-30 18:27:37 +02:00
parent 9ef8d42f0e
commit 07ffc4e7f9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -186,7 +186,7 @@ def train(cfg,
augment=True, augment=True,
hyp=hyp, # augmentation hyperparameters hyp=hyp, # augmentation hyperparameters
rect=opt.rect, # rectangular training rect=opt.rect, # rectangular training
image_weights=False) image_weights=opt.img_weights)
# Dataloader # Dataloader
dataloader = torch.utils.data.DataLoader(dataset, dataloader = torch.utils.data.DataLoader(dataset,
@ -349,6 +349,7 @@ if __name__ == '__main__':
parser.add_argument('--xywh', action='store_true', help='use xywh loss instead of GIoU loss') parser.add_argument('--xywh', action='store_true', help='use xywh loss instead of GIoU loss')
parser.add_argument('--evolve', action='store_true', help='evolve hyperparameters') parser.add_argument('--evolve', action='store_true', help='evolve hyperparameters')
parser.add_argument('--bucket', type=str, default='', help='gsutil bucket') parser.add_argument('--bucket', type=str, default='', help='gsutil bucket')
parser.add_argument('--img-weights', action='store_true', help='select training images by weight')
parser.add_argument('--var', default=0, type=int, help='debug variable') parser.add_argument('--var', default=0, type=int, help='debug variable')
opt = parser.parse_args() opt = parser.parse_args()
print(opt) print(opt)