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

Fix: colab resuming training folder name bug

Browse Source
This commit is contained in:
litagin02
2024-01-01 13:59:25 +09:00
parent a7faf52ae4
commit a5fb0bfde9
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
colab.ipynb
View File

@@ -252,9 +252,18 @@
"# 上でつけたモデル名を入力。学習を途中からする場合はきちんとモデルが保存されているフォルダ名を入力。\n", "# 上でつけたモデル名を入力。学習を途中からする場合はきちんとモデルが保存されているフォルダ名を入力。\n",
"model_name = \"your_model_name\"\n", "model_name = \"your_model_name\"\n",
"\n", "\n",
"import yaml\n",
"from webui_train import get_path\n", "from webui_train import get_path\n",
"\n", "\n",
"dataset_path, _, _, _, config_path = get_path(model_name)\n", "dataset_path, _, _, _, config_path = get_path(model_name)\n",
"with open(\"default_config.yml\", \"r\", encoding=\"utf-8\") as f:\n",
" yml_data = yaml.safe_load(f)\n",
"yml_data[\"model_name\"] = model_name\n",
"yml_data[\"dataset_path\"] = dataset_path\n",
"with open(\"config.yml\", \"w\", encoding=\"utf-8\") as f:\n",
" yaml.dump(yml_data, f, allow_unicode=True)\n",
"\n",
"\n",
"!python train_ms.py --config {config_path} --model {dataset_path}" "!python train_ms.py --config {config_path} --model {dataset_path}"
] ]
}, },