search

RetroCirce / TONet

The official implementation of "TONet: Tone-Octave Network for Singing Melody Extraction from Polyphonic Music"
https://arxiv.org/abs/2202.00951
41 stars 5 forks source link

Hey,Chen #6

Open twodog668 opened 1 year ago

twodog668 commented 1 year ago

I got a problem in testing the model like that: TypeError : test_step() missing 1 required positional argument: 'dataset_idx', and I can't fix it. Could you please help me fix it? This problem is in model/tonet.py, 454 row. Thank you very much.

RetroCirce commented 10 months ago

Hi,

I once used the dataset_idx/dataloader_idx because I test multiple test sets/validation sets when training the model. I.e., after I train 1 epoch, I test each validation sets (namely idx 1, idx 2, idx 3, etc.) during the validation stage.

I think the update of pytorch lightning removes this parameters. I suggest using the pytorch lightning 1.8.1 version