'CNN_MLP' object has no attribute 'train'

kimhc6028 / relational-networks

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

https://arxiv.org/pdf/1706.01427.pdf

BSD 3-Clause "New" or "Revised" License

811 stars 162 forks source link

'CNN_MLP' object has no attribute 'train' #23

Closed RishikMani closed 5 years ago

RishikMani commented 5 years ago

I am implementing something using CLEVR dataset and is using your repository for reference purpose.

But everytime I am executing the script, I am getting the error _'CNNMLP' object has no attribute 'train'. In main.py right after the function 'train' starts, it executes model.train(). Somehow I cannot go beyond this point.

Could somebody help me to understand if I am doing something wrong?