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[bot]]

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.