tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Improve: when initialize, if models dir exists, backup and delete

Browse Source
This commit is contained in:
litagin02
2024-01-12 09:05:28 +09:00
parent 6f1707b11a
commit 1dc4e45480
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
webui_train.py
View File

@@ -60,13 +60,18 @@ def initialize(model_name, batch_size, epochs, save_every_steps, bf16_run):
config["train"]["eval_interval"] = save_every_steps config["train"]["eval_interval"] = save_every_steps
model_path = os.path.join(dataset_path, "models") model_path = os.path.join(dataset_path, "models")
if os.path.exists(model_path):
logger.warning(f"Step 1: {model_path} already exists, so copy it to backup.")
shutil.copytree(
src=model_path,
dst=os.path.join(dataset_path, "models_backup"),
)
shutil.rmtree(model_path)
try: try:
shutil.copytree( shutil.copytree(
src="pretrained", src="pretrained",
dst=model_path, dst=model_path,
) )
except FileExistsError:
logger.warning(f"Step 1: {model_path} already exists.")
except FileNotFoundError: except FileNotFoundError:
logger.error("Step 1: `pretrained` folder not found.") logger.error("Step 1: `pretrained` folder not found.")
return False, "Step 1, Error: pretrainedフォルダが見つかりません。" return False, "Step 1, Error: pretrainedフォルダが見つかりません。"