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_datasets/densenet_rs.py: 100%

1#!/usr/bin/env python 

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

3 

4 

5"""CNN for radiological findings detection 

6 

7A Densenet121 model for radiological extraction 

8 

9""" 

10 

11import torch 

12import torchvision.models as models 

13from torch.optim import Adam 

14from torch.nn import BCEWithLogitsLoss 

15import torch.nn as nn 

16from ...models.densenet_rs import build_densenetrs 

17 

18# Import the default protocol if none is available 

19if 'dataset' not in locals(): 

20 from ..datasets.nih_cxr14_re.default import dataset 

21 

22##### Config ##### 

23lr = 1e-4 

24 

25# model 

26model = build_densenetrs() 

27 

28# optimizer 

29optimizer = Adam( 

30 filter(lambda p: p.requires_grad, model.model.model_ft.parameters()), 

31 lr=lr) 

32 

33# criterion 

34criterion = BCEWithLogitsLoss() 

35criterion_valid = BCEWithLogitsLoss()