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

1#!/usr/bin/env python 

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

3 

4 

5"""DenseNet""" 

6 

7from torch.optim import Adam 

8from torch.nn import BCEWithLogitsLoss 

9from ...models.densenet import build_densenet 

10 

11 

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

13lr = 0.0001 

14 

15# model 

16model = build_densenet(pretrained=False) 

17 

18# optimizer 

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

20 

21# criterion 

22criterion = BCEWithLogitsLoss()