From 31d96ad5589479c2acb45aa0e9df9689ed0330c8 Mon Sep 17 00:00:00 2001 From: shayaharon Date: Mon, 5 Jun 2023 15:26:38 +0300 Subject: [PATCH] strict_load fixed in recipe default checkpoint_params --- .../recipes/checkpoint_params/default_checkpoint_params.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/super_gradients/recipes/checkpoint_params/default_checkpoint_params.yaml b/src/super_gradients/recipes/checkpoint_params/default_checkpoint_params.yaml index 25036d81c8..513c565f0b 100644 --- a/src/super_gradients/recipes/checkpoint_params/default_checkpoint_params.yaml +++ b/src/super_gradients/recipes/checkpoint_params/default_checkpoint_params.yaml @@ -5,5 +5,5 @@ external_checkpoint_path: # checkpoint path that is not located in super_gradien source_ckpt_folder_name: # dirname for checkpoint loading strict_load: # key matching strictness for loading checkpoint's weights _target_: super_gradients.training.sg_trainer.StrictLoad - value: True + value: no_key_matching pretrained_weights: # a string describing the dataset of the pretrained weights (for example "imagenent").