Quellcode durchsuchen

Fix: evolve with wandb (#2634)

5.0
Ayush Chaurasia GitHub vor 3 Jahren
Ursprung
Commit
dc51e80b00
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. +1
    -1
      train.py

+ 1
- 1
train.py Datei anzeigen

@@ -439,7 +439,7 @@ def train(hyp, opt, device, tb_writer=None):
strip_optimizer(f) # strip optimizers
if opt.bucket:
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',
name='run_' + wandb_logger.wandb_run.id + '_model',
aliases=['last', 'best', 'stripped'])

Laden…
Abbrechen
Speichern