BYOL object has no attribute module

[dekaartina]

Hello , I am getting an error in the first step. Plz guide me to solve this error .

File "tools/train.py", line 142, in main() File "tools/train.py", line 132, in main train_model( File "/data2/dl/grp20/ORL/openselfsup/apis/train.py", line 97, in train_model _dist_train( File "/data2/dl/grp20/ORL/openselfsup/apis/train.py", line 233, in _dist_train runner.run(data_loaders, cfg.workflow, cfg.total_epochs) File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/epoch_based_runner.py", line 122, in run epoch_runner(data_loaders[i], **kwargs) File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/epoch_based_runner.py", line 29, in train self.call_hook('before_train_iter') File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/mmcv/runner/base_runner.py", line 282, in call_hook getattr(hook, fn_name)(self) File "/data2/dl/grp20/ORL/openselfsup/hooks/byol_hook.py", line 26, in before_train_iter assert hasattr(runner.model.module, 'momentum'),
File "/data2/dl/grp20/grp20_env/lib/python3.8/site-packages/torch/nn/modules/module.py", line 947, in getattr raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'BYOL' object has no attribute 'module'