bob.med.tb.configs.models.alexnet

AlexNet

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


"""AlexNet"""

from torch.optim import SGD
from torch.nn import BCEWithLogitsLoss
from ...models.alexnet import build_alexnet


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

# model
model = build_alexnet(pretrained=False)

# optimizer
optimizer = SGD(model.parameters(), lr=lr, momentum=0.1)

# criterion
criterion = BCEWithLogitsLoss()