File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 66
77
88def test_from_pretrained_with_mismatched_keys ():
9- orginal_model = smp .Unet (classes = 1 )
9+ original_model = smp .Unet (classes = 1 )
1010
1111 with tempfile .TemporaryDirectory () as temp_dir :
12- orginal_model .save_pretrained (temp_dir )
12+ original_model .save_pretrained (temp_dir )
1313
1414 # we should catch warning here and check if there specific keys there
1515 with pytest .warns (UserWarning ):
@@ -18,7 +18,7 @@ def test_from_pretrained_with_mismatched_keys():
1818 assert restored_model .segmentation_head [0 ].out_channels == 2
1919
2020 # verify all the weight are the same expect mismatched ones
21- original_state_dict = orginal_model .state_dict ()
21+ original_state_dict = original_model .state_dict ()
2222 restored_state_dict = restored_model .state_dict ()
2323
2424 expected_mismatched_keys = [
You can’t perform that action at this time.
0 commit comments