**_config in train.py optimizer definition

[ruohuali]

Hello
Why in train.py's optimizer definition **_config is used instead of _config?
optimizer = torch.optim.SGD(model.parameters(), **_config['optim'])
This caused error when I tried to run train.py
optimizer = torch.optim.SGD(model.parameters(), **_config['optim']) KeyError: 'optim'
Thx for the help!

[DpDark]

Hello, I have the same problem. How did you solve it？

[2679622694]

I have the same question

[sharmaannapurna]

HI, I am getting the same error when I tried to run the code. Any workaround?

[ruohuali]

I vaguely recall that it was a mistake about using wrong setup in config.py (I used test when it should be train or something like that).