Titles for check commands (did I test this? no?)

This commit is contained in:
Brandon Thomas
2023-01-12 03:41:14 -05:00
parent e3cc57979f
commit ca28449b1b
2 changed files with 8 additions and 1 deletions
+4
View File
@@ -17,6 +17,10 @@ from hifigan.env import AttrDict
from hifigan.models import Generator
from hifigan.denoiser import Denoiser as HifiDenoiser
from utils import print_title
print_title('Check Hifigan Model (Tacotron)')
parser = argparse.ArgumentParser(description='Check that the HiFi-GAN model is valid')
parser.add_argument('--checkpoint_path', type=str, help='path the hifigan model', required=True)
@@ -19,6 +19,10 @@ sys.path.append("./hifigan/")
from hifigan.generator import HifiganGenerator
from hifigan.generator import consume_prefix_in_state_dict_if_present
from utils import print_title
print_title('Check Hifigan Model (SoftVC)')
from gpu_utils import print_gpu_info
print_gpu_info()
@@ -69,4 +73,3 @@ metadata = {
with open(args.output_metadata_filename, 'w') as json_file:
json.dump(metadata, json_file)