Commit 1a12ac75 authored by dreamerlin's avatar dreamerlin
Browse files

fix

parent e70dca8f
...@@ -58,7 +58,7 @@ class Conv2d(nn.Conv2d): ...@@ -58,7 +58,7 @@ class Conv2d(nn.Conv2d):
return super().forward(x) return super().forward(x)
@CONV_LAYERS.register_module('Conv', force=True) @CONV_LAYERS.register_module('Conv3d', force=True)
class Conv3d(nn.Conv3d): class Conv3d(nn.Conv3d):
def forward(self, x): def forward(self, x):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment