Commit 328192ce authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

changed classifier block parent module: nn.Module

parent fb9b7bb2
......@@ -248,7 +248,7 @@ class DecoderBlock(ConvolutionalBlock):
return Y
class ClassifierBlock(ConvolutionalBlock):
class ClassifierBlock(nn.Module):
"""Classifier block for a U-net.
This class creates a simple classifier block following the architecture:
......
Supports Markdown
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