From 88e12dcbb0b11120b0cc90c187501e6c114b385c Mon Sep 17 00:00:00 2001 From: Eldar Kurtic Date: Tue, 27 Apr 2021 11:18:07 +0200 Subject: [PATCH] Fix for integrations/timm checkpoint path This PR fixes issue https://github.com/neuralmagic/sparseml/issues/197 --- integrations/timm/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/timm/train.py b/integrations/timm/train.py index f4c155aab1d..4e9c8c5c028 100755 --- a/integrations/timm/train.py +++ b/integrations/timm/train.py @@ -389,7 +389,7 @@ def main(): elif args.initial_checkpoint.startswith("zoo:"): # Load weights from a SparseZoo model stub zoo_model = Zoo.load_model_from_stub(args.initial_checkpoint) - args.initial_checkpoint = zoo_model.download_framework_files(extensions=[".pth"]) + args.initial_checkpoint = zoo_model.download_framework_files(extensions=[".pth"])[0] #################################################################################### # End - SparseML optional load weights from SparseZoo ####################################################################################