Commit d3965b94 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Fix typo

parent 7e68928f
......@@ -16,7 +16,7 @@ class CiompiDO(nn.Module):
if modality == 'CT/PET':
assert n_channels == 2
else:
assert n_channels == 1:
assert n_channels == 1
super(CiompiDO, self).__init__()
self.n_classes = n_classes
......
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