@@ -39,7 +39,7 @@ class PretrainedTests(unittest.TestCase):
with self.mem_file() as f:
npz.save_npz(f, model)
f.seek(0)
- new_model.load_for_inference(f, n_classes=200)
+ new_model.load_for_finetune(f, n_classes=200)
self.assertTrue(test_utils.is_all_equal(model, new_model, strict=True))
