The trunk network train with classification method, and num_spkr defines the number of the classes. Add:
args.num_spkr = 5994
will resolve that error. But it’s not straightforward process to run the code without downloading and pre processing the data, so I’ll write a snippet and inform you soon.