[F] Update test for BACKBONES.build({name: timm_resnet50})

DocsaidLab · Dec 24, 2024 · 7edf8f7 · 7edf8f7
1 parent 9a125a8
commit 7edf8f7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/tools/test_calflops.py b/tests/tools/test_calflops.py
@@ -5,8 +5,7 @@
 
 
 def test_calcualte_flops():
-    timm_create = BACKBONES.build({'name': 'timm_resnet50'})
-    model = timm_create(model_name='resnet50')
+    model = BACKBONES.build({'name': 'timm_resnet50'})
     flops, macs, params = calculate_flops(model, (1, 3, 224, 224))
     assert flops == '8.21 GFLOPS'
     assert macs == '4.09 GMACs'