Updated comments

Author: khornlund

tests/test_model.py

@@ -80,8 +80,8 @@ def test_modify_dropout(net, img_size):
 
 
 @pytest.mark.parametrize('img_size', [224, 256, 512])
-def test_modify_norm(net, img_size):
-    """Test ability to modify norm layer of network"""
+def test_modify_pool(net, img_size):
+    """Test ability to modify pooling module of network"""
 
     class AdaptiveMaxAvgPool(nn.Module):
 
@@ -104,4 +104,4 @@ def forward(self, x):
 
     data = torch.zeros((2, 3, img_size, img_size))
     output = net(data)
-    assert not torch.isnan(output).any()
+    assert not torch.isnan(output).any()