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

Upload all Data/ to hub

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

18
train_ms.py
View File

@@ -171,6 +171,16 @@ def run():
os.path.join(config.out_dir, "style_vectors.npy"), 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.manual_seed(hps.train.seed)
torch.cuda.set_device(local_rank) torch.cuda.set_device(local_rank)
@@ -489,8 +499,8 @@ def run():
if hps.repo_id is not None: if hps.repo_id is not None:
future1 = api.upload_folder( future1 = api.upload_folder(
repo_id=hps.repo_id, repo_id=hps.repo_id,
folder_path=model_dir, folder_path=config.dataset_path,
path_in_repo=f"Data/{config.model_name}/models", path_in_repo=f"Data/{config.model_name}",
delete_patterns="*.pth", delete_patterns="*.pth",
run_as_future=True, run_as_future=True,
) )
@@ -790,8 +800,8 @@ def train_and_evaluate(
if hps.repo_id is not None: if hps.repo_id is not None:
api.upload_folder( api.upload_folder(
repo_id=hps.repo_id, repo_id=hps.repo_id,
folder_path=hps.model_dir, folder_path=config.dataset_path,
path_in_repo=f"Data/{config.model_name}/models", path_in_repo=f"Data/{config.model_name}",
delete_patterns="*.pth", delete_patterns="*.pth",
run_as_future=True, run_as_future=True,
) )