model = _load_state(cls, checkpoint, strict=strict, **kwargs)
File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/pytorch_lightning/core/saving.py", line 158, in _load_state
obj = cls(**_cls_kwargs)
File "/teamspace/studios/this_studio/lang-seg/modules/lseg_module.py", line 55,
File "/home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/pytorch_lightning/core/saving.py", line 158, in _load_state obj = cls(**_cls_kwargs) File "/teamspace/studios/this_studio/lang-seg/modules/lseg_module.py", line 55,