lRomul/argus-freesound

val_step() takes 2 positional arguments but 3 were given when training

Closed this issue · 5 comments

I run the train_fold.py and meet the following problem when training.
val_step() takes 2 positional arguments but 3 were given
I have no idea what leads to this problem. In each step, we get a tri-tuple from the dataloader, namely image, target, noisy. But the noisy is a zero tensor, how to feeds the noisy tenor to model ? Dose the noisy tensor result in the problem?

I locate the problem with step debug
image
Everything seems be OK, but it reports that val_step() takes 2 positional arguments but 3 were given
image

Hi! Thank you for reporting the problem and additional information. Today I will try to reproduce the error.

I successfully run training via instructions in the readme.
I need more information. What environment do you have? Do you use Docker or your setup up? If yours which versions of python and packages do you use?

I successfully run training via instructions in the readme.
I need more information. What environment do you have? Do you use Docker or your setup up? If yours which versions of python and packages do you use?

Thanks for your reply. I just found that I install the newest pytorch-argus but I didn't notice that you updated the argus_models.py several month ago. After changing the code in argus_models.py, it run successfully, Thanks again.

Great, glad the issue is solved!