|
|
|
@ -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",
|
|
|
|
|