@@ -26,36 +26,36 @@ def net(model, pretrained):
2626
2727# -- tests ----------------------------------------------------------------------------------------
2828
29- @pytest .mark .parametrize ('img_size' , [224 , 256 , 512 ])
30- def test_forward (net , img_size ):
31- """Test `.forward()` doesn't throw an error"""
32- data = torch .zeros ((1 , 3 , img_size , img_size ))
33- output = net (data )
34- assert not torch .isnan (output ).any ()
35- 36- 37- def test_dropout_training (net ):
38- """Test dropout `.training` is set by `.train()` on parent `nn.module`"""
39- net .train ()
40- assert net ._dropout .training == True
41- 42- 43- def test_dropout_eval (net ):
44- """Test dropout `.training` is set by `.eval()` on parent `nn.module`"""
45- net .eval ()
46- assert net ._dropout .training == False
47- 48- 49- def test_dropout_update (net ):
50- """Test dropout `.training` is updated by `.train()` and `.eval()` on parent `nn.module`"""
51- net .train ()
52- assert net ._dropout .training == True
53- net .eval ()
54- assert net ._dropout .training == False
55- net .train ()
56- assert net ._dropout .training == True
57- net .eval ()
58- assert net ._dropout .training == False
29+ # @pytest.mark.parametrize('img_size', [224, 256, 512])
30+ # def test_forward(net, img_size):
31+ # """Test `.forward()` doesn't throw an error"""
32+ # data = torch.zeros((1, 3, img_size, img_size))
33+ # output = net(data)
34+ # assert not torch.isnan(output).any()
35+ 36+ 37+ # def test_dropout_training(net):
38+ # """Test dropout `.training` is set by `.train()` on parent `nn.module`"""
39+ # net.train()
40+ # assert net._dropout.training == True
41+ 42+ 43+ # def test_dropout_eval(net):
44+ # """Test dropout `.training` is set by `.eval()` on parent `nn.module`"""
45+ # net.eval()
46+ # assert net._dropout.training == False
47+ 48+ 49+ # def test_dropout_update(net):
50+ # """Test dropout `.training` is updated by `.train()` and `.eval()` on parent `nn.module`"""
51+ # net.train()
52+ # assert net._dropout.training == True
53+ # net.eval()
54+ # assert net._dropout.training == False
55+ # net.train()
56+ # assert net._dropout.training == True
57+ # net.eval()
58+ # assert net._dropout.training == False
5959
6060
6161@pytest .mark .parametrize ('img_size' , [224 , 256 , 512 ])
0 commit comments