From 6d2acec1bb1fa40fa8f65771e2f18c805920d4e3 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Sat, 2 Oct 2021 16:10:11 +0100 Subject: [PATCH] Fix ordering of tests --- tests/test_utils.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 3e11eacc..b0f890d2 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -44,17 +44,14 @@ def test_freeze_unfreeze(): assert isinstance(model.layer1[0].bn1, BatchNorm2d) assert model.layer2[0].conv1.weight.requires_grad == True - # Freeze BN + # Freeze/unfreeze BN # From root freeze(model, ['layer1.0.bn1']) assert isinstance(model.layer1[0].bn1, FrozenBatchNorm2d) - # From direct parent - freeze(model.layer1[0], ['bn1']) - assert isinstance(model.layer1[0].bn1, FrozenBatchNorm2d) - - # Unfreeze BN unfreeze(model, ['layer1.0.bn1']) assert isinstance(model.layer1[0].bn1, BatchNorm2d) # From direct parent + freeze(model.layer1[0], ['bn1']) + assert isinstance(model.layer1[0].bn1, FrozenBatchNorm2d) unfreeze(model.layer1[0], ['bn1']) assert isinstance(model.layer1[0].bn1, BatchNorm2d) \ No newline at end of file