Ayush Chaurasia
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| strip_optimizer(f) # strip optimizers | strip_optimizer(f) # strip optimizers | ||||
| if opt.bucket: | if opt.bucket: | ||||
| os.system(f'gsutil cp {final} gs://{opt.bucket}/weights') # upload | os.system(f'gsutil cp {final} gs://{opt.bucket}/weights') # upload | ||||
| if wandb_logger.wandb: # Log the stripped model | |||||
| if wandb_logger.wandb and not opt.evolve: # Log the stripped model | |||||
| wandb_logger.wandb.log_artifact(str(final), type='model', | wandb_logger.wandb.log_artifact(str(final), type='model', | ||||
| name='run_' + wandb_logger.wandb_run.id + '_model', | name='run_' + wandb_logger.wandb_run.id + '_model', | ||||
| aliases=['last', 'best', 'stripped']) | aliases=['last', 'best', 'stripped']) |
Loading…