Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor fix for pruning and moving.

This commit is contained in:
David Bielejeski 2022-09-28 15:25:00 -05:00
parent fce2d1f17d
commit 5f471dc3ab
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreambooth_runpod_joepenna.ipynb
View File

@ -442,7 +442,7 @@
"last_checkpoint_file_pruned = directory_paths[-1] + \"/checkpoints/last-pruned.ckpt\"\n",
"training_samples = !ls training_samples\n",
"date_string = !date +\"%Y-%m-%dT%H-%M-%S\"\n",
"file_name = date_string[-1] + \"_\" + project_name + \"_\" + str(len(training_samples)) + \"_training_images_\" + str(batch_size) + \"_batch_size_\" + class_word + \"_class_word.ckpt\"\n",
"file_name = date_string[-1] + \"_\" + project_name + \"_\" + str(len(training_samples)) + \"_training_images_\" + str(max_steps) + \"_max_steps_\" + class_word + \"_class_word.ckpt\"\n",
"!mkdir -p trained_models\n",
"!mv {last_checkpoint_file_pruned} trained_models/{file_name}"
]