From 5b60bf705a03d3881c96a0913b1e52628c042949 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Tue, 20 Feb 2024 00:52:12 +0900 Subject: [PATCH] Upload all Data/ to hub --- train_ms.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/train_ms.py b/train_ms.py index c16dda7..51e8d61 100644 --- a/train_ms.py +++ b/train_ms.py @@ -171,6 +171,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) @@ -489,8 +499,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, ) @@ -790,8 +800,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, )