mednet.config.classify.models.cnn3d

Simple CNN for 3D image classification, to be trained from scratch.

# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""Simple CNN for 3D image classification, to be trained from scratch."""

from torch.nn import BCEWithLogitsLoss
from torch.optim import Adam

from mednet.models.classify.cnn3d import Conv3DNet

model = Conv3DNet(
    loss_type=BCEWithLogitsLoss,
    optimizer_type=Adam,
    optimizer_arguments=dict(lr=8e-5),
)