Browse Source

更新 'evaluate.py'

master
unicorn 1 month ago
parent
commit
7ce2a766cd
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      evaluate.py

+ 2
- 2
evaluate.py View File

@@ -114,7 +114,7 @@ parser.add_argument(
choices=['Ascend', 'GPU', 'CPU'],
help='device where the code will be implemented (default: Ascend)')

parser.add_argument('--ckpt_path', type=str, default="DPRNN-10_3560.ckpt",
parser.add_argument('--ckpt_path', type=str, default="DPRNN-10_445.ckpt",
help='Path to model file created by training')

parser.add_argument('--cal_sdr', type=int, default=1,
@@ -179,7 +179,7 @@ def evaluate(args):
tt_dataset = DatasetGenerator(args.train_dir, args.batch_size,
sample_rate=args.sample_rate, segment=args.segment)
tt_loader = ds.GeneratorDataset(tt_dataset, ["mixture", "lens", "sources"], shuffle=False)
tt_loader = tt_loader.batch(batch_size=2)
tt_loader = tt_loader.batch(batch_size=4)

for data in tt_loader.create_dict_iterator():
padded_mixture = data["mixture"]


Loading…
Cancel
Save