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

updates

This commit is contained in:
Glenn Jocher 2019-06-24 14:46:00 +02:00
parent a3fcf20385
commit d208f006a1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -292,7 +292,7 @@ if __name__ == '__main__':
parser.add_argument('--rank', default=0, type=int, help='distributed training node rank') parser.add_argument('--rank', default=0, type=int, help='distributed training node rank')
parser.add_argument('--world-size', default=1, type=int, help='number of nodes for distributed training') parser.add_argument('--world-size', default=1, type=int, help='number of nodes for distributed training')
parser.add_argument('--backend', default='nccl', type=str, help='distributed backend') parser.add_argument('--backend', default='nccl', type=str, help='distributed backend')
parser.add_argument('--nosave', action='store_true', help='do not save training results') parser.add_argument('--nosave', action='store_true', help='only save final checkpoint')
parser.add_argument('--notest', action='store_true', help='only test final epoch') parser.add_argument('--notest', action='store_true', help='only test final epoch')
parser.add_argument('--giou', action='store_true', help='use GIoU loss instead of xy, wh loss') parser.add_argument('--giou', action='store_true', help='use GIoU loss instead of xy, wh loss')
parser.add_argument('--evolve', action='store_true', help='run hyperparameter evolution') parser.add_argument('--evolve', action='store_true', help='run hyperparameter evolution')
@ -301,8 +301,8 @@ if __name__ == '__main__':
print(opt) print(opt)
if opt.evolve: if opt.evolve:
opt.notest = True # save time by only testing final epoch opt.notest = True # only test final epoch
opt.nosave = True # do not save checkpoints opt.nosave = True # only save final checkpoint
# Train # Train
results = train( results = train(