about code issues

[andre20000131]

hi,
I found some issues about your code..first,you don't upload complete files, for example: solver.py and models and so on...
And,i found your code where on main.py 62 line, you should define [1,2,3] ... not a string... you don't upload the way how to train and test,,if it is necessary pls upload them...thx...

[andre20000131]

again issues, when I was trainings found some code wrong,MAE_illum at train model should be MAE_illum[0].item() not item()...