|
|
@@ -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))
|
|
|
|
