wwu-mmll / deepbet

Fast brain extraction using neural networks
MIT License
24 stars 3 forks source link

Input Error #3

Closed qingfengxulai199 closed 1 month ago

qingfengxulai199 commented 1 month ago

_20240515211746 NotImplementedError: Input Error: Only 3D, 4D and 5D input Tensors supported (got 5D) for the modes: nearest | linear | bilinear | bicubic | trilinear (got neare st-exact)

Would you have any approaches to solve it? Your early reply would be highly appreciated.

codingfisch commented 1 month ago

The "nearest-exact" interpolation mode requires torch version >= 1.9.0. I will add this dependency requirement in an upcoming release!

qingfengxulai199 commented 1 month ago

Thank you for your answer, but I still have this problem after upgrading the torch Snipaste_2024-05-21_21-24-41

codingfisch commented 1 month ago

Sorry, you are right. I have tested it myself and pytorch>=1.11.0 is needed!

qingfengxulai199 commented 1 month ago

Thank you for your help.