Simplify example training

This commit is contained in:
Augusto de la Torre 2023-01-24 01:12:56 +01:00
parent 158dea22da
commit 9bb213efaf
1 changed files with 2 additions and 4 deletions

View File

@ -217,9 +217,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"project_name = \"ft_v1\"\n", "!python train.py --project_name \"ft_v1a_512_1e6\" \\\n",
"\n",
"!python train.py --project_name \"ft_v1_512_1e6\" \\\n",
"--resume_ckpt \"{ckpt_name}\" \\\n", "--resume_ckpt \"{ckpt_name}\" \\\n",
"--data_root \"input\" \\\n", "--data_root \"input\" \\\n",
"--resolution 512 \\\n", "--resolution 512 \\\n",
@ -253,7 +251,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"\n", "\n",
"!python train.py --project_name \"{project_name}_512_7e7\" \\\n", "!python train.py --project_name \"ft_v1b_512_7e7\" \\\n",
"--resume_ckpt findlast \\\n", "--resume_ckpt findlast \\\n",
"--data_root \"input\" \\\n", "--data_root \"input\" \\\n",
"--resolution 512 \\\n", "--resolution 512 \\\n",