mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

W&B evolution bug fix (#1373)

pull/1350/head
Glenn Jocher 2020-11-12 12:55:52 +01:00 committed by GitHub
parent f6b3c96966
commit ce3872f86c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -323,7 +323,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
dataloader=testloader, dataloader=testloader,
save_dir=log_dir, save_dir=log_dir,
plots=epoch == 0 or final_epoch, # plot first and last plots=epoch == 0 or final_epoch, # plot first and last
log_imgs=opt.log_imgs) log_imgs=opt.log_imgs if wandb else 0)
# Write # Write
with open(results_file, 'a') as f: with open(results_file, 'a') as f:
@ -475,7 +475,6 @@ if __name__ == '__main__':
assert os.environ.get('WANDB_DISABLED') != 'true' assert os.environ.get('WANDB_DISABLED') != 'true'
except (ImportError, AssertionError): except (ImportError, AssertionError):
opt.log_imgs = 0
logger.info("Install Weights & Biases for experiment logging via 'pip install wandb' (recommended)") logger.info("Install Weights & Biases for experiment logging via 'pip install wandb' (recommended)")
train(hyp, opt, device, tb_writer, wandb) train(hyp, opt, device, tb_writer, wandb)