| 455 | |
| 456 | # but3.click(click_train,[exp_dir1,sr2,if_f0_3,save_epoch10,total_epoch11,batch_size12,if_save_latest13,pretrained_G14,pretrained_D15,gpus16]) |
| 457 | def click_train( |
| 458 | exp_dir1, |
| 459 | sr2, |
| 460 | if_f0_3, |
| 461 | spk_id5, |
| 462 | save_epoch10, |
| 463 | total_epoch11, |
| 464 | batch_size12, |
| 465 | if_save_latest13, |
| 466 | pretrained_G14, |
| 467 | pretrained_D15, |
| 468 | gpus16, |
| 469 | if_cache_gpu17, |
| 470 | if_save_every_weights18, |
| 471 | version19, |
| 472 | ): |
| 473 | # 生成filelist |
| 474 | exp_dir = "%s/logs/%s" % (now_dir, exp_dir1) |
| 475 | os.makedirs(exp_dir, exist_ok=True) |
| 476 | gt_wavs_dir = "%s/0_gt_wavs" % (exp_dir) |
| 477 | feature_dir = ( |
| 478 | "%s/3_feature256" % (exp_dir) |
| 479 | if version19 == "v1" |
| 480 | else "%s/3_feature768" % (exp_dir) |
| 481 | ) |
| 482 | if if_f0_3: |
| 483 | f0_dir = "%s/2a_f0" % (exp_dir) |
| 484 | f0nsf_dir = "%s/2b-f0nsf" % (exp_dir) |
| 485 | names = ( |
| 486 | set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) |
| 487 | & set([name.split(".")[0] for name in os.listdir(feature_dir)]) |
| 488 | & set([name.split(".")[0] for name in os.listdir(f0_dir)]) |
| 489 | & set([name.split(".")[0] for name in os.listdir(f0nsf_dir)]) |
| 490 | ) |
| 491 | else: |
| 492 | names = set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) & set( |
| 493 | [name.split(".")[0] for name in os.listdir(feature_dir)] |
| 494 | ) |
| 495 | opt = [] |
| 496 | for name in names: |
| 497 | if if_f0_3: |
| 498 | opt.append( |
| 499 | "%s/%s.wav|%s/%s.npy|%s/%s.wav.npy|%s/%s.wav.npy|%s" |
| 500 | % ( |
| 501 | gt_wavs_dir.replace("\\", "\\\\"), |
| 502 | name, |
| 503 | feature_dir.replace("\\", "\\\\"), |
| 504 | name, |
| 505 | f0_dir.replace("\\", "\\\\"), |
| 506 | name, |
| 507 | f0nsf_dir.replace("\\", "\\\\"), |
| 508 | name, |
| 509 | spk_id5, |
| 510 | ) |
| 511 | ) |
| 512 | else: |
| 513 | opt.append( |
| 514 | "%s/%s.wav|%s/%s.npy|%s" |