Hugging Face, Weights And Biases, Fiftyone, Model Evaluation, Hyperparameter Tuning.
device) # Log training metrics in W&B wandb.log({ "epoch": epoch, "lr": optimizer.defaults["lr"], "train_loss": train_loss, "val_loss": val_loss }) if epoch % save_epoch == 0: model.save_pretrained( os.path.join(output_folder,str(epoch)) ) model.save_pretrained(os.path.join(output_folder, "final"))