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

Fix and upload all Data/ to hub

Browse Source
This commit is contained in:
litagin02
2024-02-20 00:42:00 +09:00
parent ea28ec9058
commit a04e569c86
2 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
train_ms_jp_extra.py
View File

@@ -175,6 +175,16 @@ def run():
os.path.join(config.out_dir, "style_vectors.npy"),
)
if args.repo_id is not None:
# Upload Dataset
api.upload_folder(
repo_id=hps.repo_id,
folder_path=config.dataset_path,
path_in_repo=f"Data/{config.model_name}",
delete_patterns="*.pth",
run_as_future=True,
)
torch.manual_seed(hps.train.seed)
torch.cuda.set_device(local_rank)
@@ -577,8 +587,8 @@ def run():
if hps.repo_id is not None:
future1 = api.upload_folder(
repo_id=hps.repo_id,
folder_path=model_dir,
path_in_repo=f"Data/{config.model_name}/models",
folder_path=config.dataset_path,
path_in_repo=f"Data/{config.model_name}",
delete_patterns="*.pth",
run_as_future=True,
)
@@ -947,8 +957,8 @@ def train_and_evaluate(
if hps.repo_id is not None:
api.upload_folder(
repo_id=hps.repo_id,
folder_path=hps.model_dir,
path_in_repo=f"Data/{config.model_name}/models",
folder_path=config.dataset_path,
path_in_repo=f"Data/{config.model_name}",
delete_patterns="*.pth",
run_as_future=True,
)