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

1#!/usr/bin/env python 

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

3 

4 

5"""AlexNet 

6 

7Pretrained AlexNet 

8""" 

9 

10from torch.optim import SGD 

11from torch.nn import BCEWithLogitsLoss 

12from ...models.alexnet import build_alexnet 

13 

14 

15##### Config ##### 

16lr = 0.001 

17 

18# model 

19model = build_alexnet(pretrained=True) 

20 

21# optimizer 

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

23 

24# criterion 

25criterion = BCEWithLogitsLoss()