AttributeError: 'NoneType' object has no attribute 'mem'
When i tried to run python main.py --cfg configs/GPS/actor-GPS.yaml wandb.use False, it came out the following error:
register_act('swish', partial(SWISH, inplace=cfg.mem.inplace))
AttributeError: 'NoneType' object has no attribute 'mem'
Sorry for getting back to you so late.
I get the same error if pytorch_lightning is not installed. This should be indicated somewhere in the stack trace of this error. Can you check your entire stack trace to see whether you get the same warning about pytorch_lightning? Installing with pip install pytorch_lightning fixed the issue on my side.
pip install pytorch_lightning would possibly solve the problem.
After I try pip install pytorch_lightning it still has this problem, can it run in the docker continer?
The environment cannot detect the named yacs module, so this code pip install yacs would possibly solve the problem