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

Merge remote-tracking branch 'origin/master'

This commit is contained in:
Glenn Jocher 2019-08-01 00:10:10 +02:00
parent 3f9521bc08 7b3d9f02ec
commit 3b694fc8d0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -131,7 +131,7 @@ def train(cfg,
optimizer.load_state_dict(chkpt['optimizer']) optimizer.load_state_dict(chkpt['optimizer'])
best_fitness = chkpt['best_fitness'] best_fitness = chkpt['best_fitness']
if chkpt['training_results'] is not None: if chkpt.get('training_results') is not None:
with open('results.txt', 'w') as file: with open('results.txt', 'w') as file:
file.write(chkpt['training_results']) # write results.txt file.write(chkpt['training_results']) # write results.txt
@ -350,7 +350,7 @@ if __name__ == '__main__':
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('--img-weights', action='store_true', help='select training images by weight')
parser.add_argument('--var', default=0, type=int, help='debug variable')
opt = parser.parse_args() opt = parser.parse_args()
print(opt) print(opt)