You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
victormay
changed the title
[Bug] The forward function of the ImageClassifier class, when the mode is set to "tensor", will pass through the neck twice.
[Bug] my bad
Aug 28, 2024
Branch
main branch (mmpretrain version)
Describe the bug
Environment
{'sys.platform': 'win32', 'Python': '3.10.10 (tags/v3.10.10:aad5f6a, Feb 7 2023, 17:20:36) [MSC v.1929 ' '64 bit (AMD64)]', 'CUDA available': True, 'MUSA available': False, 'numpy_random_seed': 2147483648, 'GPU 0': 'NVIDIA GeForce GTX 1080 Ti', 'CUDA_HOME': 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.1', 'NVCC': 'Cuda compilation tools, release 12.1, V12.1.66', 'MSVC': '用于 x64 的 Microsoft (R) C/C++ 优化编译器 19.29.30147 版', 'GCC': 'n/a', 'PyTorch': '2.2.2+cu121', 'TorchVision': '0.17.2+cu121', 'OpenCV': '4.9.0', 'MMEngine': '0.10.4', 'MMCV': '2.1.0', 'MMPreTrain': '1.2.0+'}
Other information
No response
The text was updated successfully, but these errors were encountered: