Don't compile model in classification tests

constantinpape · May 22, 2023 · d571734 · d571734
1 parent 9379657
commit d571734
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_classification.py b/test/test_classification.py
@@ -51,7 +51,8 @@ def test_classification_2d(self):
         trainer.fit(n_iterations)
 
         self._check_checkpoint(
-            "./checkpoints/test-model-2d/latest.pt", 18, trainer.model, resnet18, num_classes=n_classes
+            "./checkpoints/test-model-2d/latest.pt", 18, trainer.model, resnet18, num_classes=n_classes,
+            compile_model=False,
         )
 
     def test_classification_3d(self):
@@ -72,6 +73,7 @@ def test_classification_3d(self):
         model = resnet3d_18(in_channels=1, out_channels=n_classes)
         trainer = default_classification_trainer(
             name="test-model-3d", model=model, train_loader=loader, val_loader=loader,
+            compile_model=False,
         )
         trainer.fit(12)