bob.med.tb.configs.models.densenet_pretrained

DenseNet

#!/usr/bin/env python
# -*- coding: utf-8 -*-


"""DenseNet"""

from torch.optim import Adam
from torch.nn import BCEWithLogitsLoss
from ...models.densenet import build_densenet


##### Config #####
lr = 0.01

# model
model = build_densenet(pretrained=True)

# optimizer
optimizer = Adam(model.parameters(), lr=lr)

# criterion
criterion = BCEWithLogitsLoss()