Implement 2D and 3D global aggregation block of Non local U-Nets in pytorch.
尝试了一下,非常占显存。
修正一个错误:并不能任意形状,忽视了求导的问题。
Inplement 2D global aggregation block of Non-local-U-Nets. Written in pytorch.
PythonMIT
Implement 2D and 3D global aggregation block of Non local U-Nets in pytorch.
尝试了一下,非常占显存。
修正一个错误:并不能任意形状,忽视了求导的问题。