Docker image is not working, torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'

[thomasgauvin]

Before submitting a bug report, please be aware that your issue must be reproducible with all of the following, otherwise it is non-actionable, and we can not help you:

• Current repo: run git fetch && git status -uno to check and git pull to update repo
• Common dataset: coco.yaml or coco128.yaml
• Common environment: Colab, Google Cloud, or Docker image. See https://github.com/ultralytics/yolov5#environments

If this is a custom dataset/training question you must include your train*.jpg, test*.jpg and results.png figures, or we can not help you. You can generate these with utils.plot_results().

🐛 Bug

A clear and concise description of what the bug is.
Attempting to run the ultralytics/yolov5:latest docker image results in torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'

To Reproduce (REQUIRED)

run python detect.py ...

Output:

  File "/root/Video-Streaming/computer_vision/yolov5_mod/models/yolo.py", line 118, in forward
    return self.forward_once(x, profile)  # single-scale inference, train
  File "/root/Video-Streaming/computer_vision/yolov5_mod/models/yolo.py", line 138, in forward_once
    x = m(x)  # run
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/root/Video-Streaming/computer_vision/yolov5_mod/models/common.py", line 92, in forward
    return self.conv(torch.cat([x[..., ::2, ::2], x[..., 1::2, ::2], x[..., ::2, 1::2], x[..., 1::2, 1::2]], 1))
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/root/Video-Streaming/computer_vision/yolov5_mod/models/common.py", line 36, in fuseforward
    return self.act(self.conv(x))
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 438, in forward
    return F.hardswish(input, self.inplace)
  File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 779, in __getattr__
    type(self).__name__, name))
torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'

Expected behavior

It should run and begin analyzing the video frames

Environment

If applicable, add screenshots to help explain your problem.

• OS: Docker
• GPU 1080 Ti

[github-actions]

Hello @thomasgauvin, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook , Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

• Cloud-based AI systems operating on hundreds of HD video streams in realtime.
• Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
• Custom data training, hyperparameter evolution, and model exportation to any destination.

For more information please visit https://www.ultralytics.com.

[glenn-jocher]

Docker image operates correctly. Just pulled latest and tested today:

[glenn-jocher]

GPU mode operates correctly also:

[glenn-jocher]

See https://docs.ultralytics.com/yolov5/environments/docker_image_quickstart_tutorial/

[thomasgauvin]

Thank you for the response @glenn-jocher, I will try it again perhaps there is something wrong with my configurations

[HsLOL]

hello @thomasgauvin
I also have the same problem when I run the detect.py
The error is "torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'"
Have you solved this problem? If you solved this problem,what's your configurations?

thank you!

[littlePrince126]

@HsLOL Using torch-1.6.0 with torchvision-0.7.0 could slove this problem

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.