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'