AttributeError: module 'catalyst.dl' has no attribute 'GPUEngine'

[vaibhavi5]

🐛 Bug Report

How To Reproduce

Steps to reproduce the behavior:

1. Go to '...'
2. Run '....'
3. ...

srun: launch/slurm: launch_p_step_launch: CpuBindType=(null type)
srun: launching StepId=6688942.0 on host arctrdagn033, 1 tasks: 0
srun: route/default: init: route default plugin loaded
srun: launch/slurm: _task_start: Node arctrdagn033, 1 tasks started
Traceback (most recent call last):
File "/data/users4/vitkyal/projects/meshnet/brainsAD/babywire_train_30channels-1m.py", line 566, in
runner.run()
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 828, in run
self._run_event("on_exception")
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 751, in _run_event
getattr(self, event)(self)
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 743, in on_exception
raise self.exception
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 825, in run
self._run_experiment()
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 805, in _run_experiment
self._run_event("on_experiment_start")
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 747, in _run_event
getattr(self, event)(self)
File "/home/users/vitkyal/anaconda3/envs/torch2/lib/python3.9/site-packages/catalyst/core/runner.py", line 639, in on_experiment_start
self.engine = self.get_engine()
File "/data/users4/vitkyal/projects/meshnet/brainsAD/babywire_train_30channels-1m.py", line 192, in get_engine
return dl.GPUEngine()
AttributeError: module 'catalyst.dl' has no attribute 'GPUEngine'

Code sample

Screenshots

Expected behavior

Environment

Please copy and paste the output from our environment collection script

catalyst-contrib --collect-env
# or manually
wget https://raw.githubusercontent.com/catalyst-team/catalyst/master/catalyst/contrib/scripts/collect_env.py
python collect_env.py

(or fill out the checklist below manually).

# example checklist, fill with your info
Catalyst version: 20.04.2
PyTorch version: 1.4.0
TensorFlow version: N/A
TensorBoard version: 2.2.1
OS: Mac OSX 10.15.4
Python version: 3.7
CUDA runtime version: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA

Additional context

Checklist

• bug description
• steps to reproduce
• expected behavior
• environment
• code sample / screenshots

FAQ

Please review the FAQ before submitting an issue:

• I have read the documentation and FAQ
• I have reviewed the minimal examples section
• I have checked the changelog for main framework updates
• I have read the contribution guide
• I have joined Catalyst slack (#__questions channel) for issue discussion

[github-actions]

Hi! Thank you for your contribution! Please re-check all issue template checklists - unfilled issues would be closed automatically. And do not forget to join our slack for collaboration.