Merge pull request #33 from nawnie/patch-2

train.py
This commit is contained in:
Victor Hall 2023-01-25 10:36:50 -08:00 committed by GitHub
commit 9a9909c7ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -354,7 +354,7 @@
"if Disable_text_Encoder:\n",
" textencode = \"--disable_textenc_training Train_text \"\n",
"\n",
"!python train_colab.py --resume_ckpt \"$model\" \\\n",
"!python train.py --resume_ckpt \"$model\" \\\n",
" $textencode \\\n",
" $Gradient \\\n",
" $shuffle \\\n",
@ -378,8 +378,8 @@
" --save_every_n_epoch $Save_every_N_epoch \\\n",
" --seed 555 \\\n",
" --shuffle_tags \\\n",
" --useadam8bit \\\n",
" --notebook\n"
" --useadam8bit \n",
]
},
{
@ -464,4 +464,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}