Pytorch_lightning module : can't set attribute error

I followed this link. This should be helpful