From 40d90c868ab2d8b6f3c03f1e35e8d00dfb4d3035 Mon Sep 17 00:00:00 2001 From: Andre Anjos <andre.dos.anjos@gmail.com> Date: Mon, 1 Jul 2024 14:38:05 +0200 Subject: [PATCH] [libs.classification.config.models.cnn3d] Fix qa --- src/mednet/libs/classification/config/models/cnn3d.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/mednet/libs/classification/config/models/cnn3d.py b/src/mednet/libs/classification/config/models/cnn3d.py index 305b5452..e0559d7d 100644 --- a/src/mednet/libs/classification/config/models/cnn3d.py +++ b/src/mednet/libs/classification/config/models/cnn3d.py @@ -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, -- GitLab