Python error when training

[jackcolt]

Really cool stuff here. Thank you.

When running the train.py, I get the following cryptic python error. Any thoughts? I can train the LTSM models no problem using the source straight out of the repo.

Traceback (most recent call last): File "train.py", line 56, in main(args) File "train.py", line 21, in main model = PedalNet(vars(args)) File "/home/jack/PedalNetRT/model.py", line 118, in init self.hparams = hparams File "/home/jack/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1225, in setattr object.setattr(self, name, value) AttributeError: can't set attribute

[mishushakov]

hey, what command line arguments are you using with train.py?

[jackcolt]

Thanks for responding. I'm just calling with the params from the readme here in the repo.

python train.py data/ts9_test1_in_FP32.wav data/ts9_test1_out_FP32.wav

[GuitarML]

@jackcolt Haven't been able to reproduce what you're seeing, are you still getting this or has it been resolved?

[dobell-christopher]

Change line 118 in module.py from self.hparams = hparams to: self.hparams.update(hparams)

You must have a newer version of pytorch as self.hparams = hparams no longer works in the new version

[JamesJieranShen]

For proper model restoration, it also seems necessary to call self.save_hyperparameters() afterwards.