Yoshitaka16 commited on
Commit
01c43ab
·
verified ·
1 Parent(s): e3dad9c

Delete train.py

Browse files
Files changed (1) hide show
  1. train.py +0 -20
train.py DELETED
@@ -1,20 +0,0 @@
1
- # train.py
2
- import argparse
3
- from models.djcm.djcm_model import DJCMModel
4
-
5
- parser = argparse.ArgumentParser()
6
- parser.add_argument("--model_type", type=str, default="rvc", choices=["rvc", "djcm"])
7
- parser.add_argument("--train_data", type=str, required=True)
8
- parser.add_argument("--output_path", type=str, required=True)
9
- args = parser.parse_args()
10
-
11
- # Load model
12
- if args.model_type == "djcm":
13
- model = DJCMModel()
14
- else:
15
- from models.rvc.rvc_model import RVCModel
16
- model = RVCModel()
17
-
18
- # Dummy training loop
19
- print(f"Training {args.model_type} model with data at {args.train_data}")
20
- # Tambahkan training sebenarnya sesuai dataset RVC