Some fixes for training notebook.

pull/88/head
Mateo Cedillo 1 year ago
parent 9b58ef1406
commit 2a9d86a902

@ -364,6 +364,8 @@
"#@markdown # <font color=\"pink\"> **5. Run the TensorBoard extension.** 📈\n",
"\n",
"#@markdown The TensorBoard is used to visualize the results of the model while it is being trained.\n",
"\n",
"#@markdown **Note: due to piper update, the tensorboard is not working at the moment.**\n",
"%load_ext tensorboard\n",
"%tensorboard --logdir {output_dir}"
],
@ -389,7 +391,6 @@
" --num-test-examples 2 \\\n",
" --quality {quality} \\\n",
" --checkpoint-epochs {checkpoint_epochs} \\\n",
" --check_val_every_n_epoch {checkpoint_epochs} \\\n",
" --log_every_n_steps {log_every_n_steps} \\\n",
" --max_epochs {max_epochs} \\\n",
" {ft_command}\\\n",

Loading…
Cancel
Save