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

1#!/usr/bin/env python 

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

3 

4 

5"""CNN for Tuberculosis Detection 

6 

7Implementation of the model architecture proposed by F. Pasa in the article 

8"Efficient Deep Network Architectures for Fast Chest X-Ray Tuberculosis  

9Screening and Visualization". 

10 

11Reference: [PASA-2019]_ 

12""" 

13 

14from torch.optim import Adam 

15from torch.nn import BCEWithLogitsLoss 

16from ...models.pasa import build_pasa 

17 

18 

19##### Config ##### 

20lr = 8e-5 

21 

22# model 

23model = build_pasa() 

24 

25# optimizer 

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

27 

28# criterion 

29criterion = BCEWithLogitsLoss()