Skip to content
Snippets Groups Projects
Commit 40d90c86 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.classification.config.models.cnn3d] Fix qa

parent 9726f356
No related branches found
No related tags found
1 merge request!46Create common library
Pipeline #89237 failed
......@@ -3,11 +3,10 @@
# SPDX-License-Identifier: GPL-3.0-or-later
"""Simple CNN for 3D organ classification, to be trained from scratch."""
from mednet.models.cnn3d import Conv3DNet
from torch.nn import BCEWithLogitsLoss
from torch.optim import Adam
from mednet.models.cnn3d import Conv3DNet
model = Conv3DNet(
loss_type=BCEWithLogitsLoss,
optimizer_type=Adam,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment