Coverage for /scratch/builds/bob/bob.med.tb/miniconda/conda-bld/bob.med.tb_1674079587905/_test_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placeho/lib/python3.10/site-packages/bob/med/tb/configs/models/alexnet.py: 100%

1#!/usr/bin/env python 

2# -*- coding: utf-8 -*- 

3 

4 

5"""AlexNet""" 

6 

7from torch.optim import SGD 

8from torch.nn import BCEWithLogitsLoss 

9from ...models.alexnet import build_alexnet 

10 

11 

12##### Config ##### 

13lr = 0.01 

14 

15# model 

16model = build_alexnet(pretrained=False) 

17 

18# optimizer 

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

20 

21# criterion 

22criterion = BCEWithLogitsLoss()