diff --git a/docs/source/tutorials/Tutorial 02 - Loading Data from Unstructured Directory.ipynb b/docs/source/tutorials/Tutorial 02 - Loading Data from Unstructured Directory.ipynb index 608be71..dfc8d1a 100644 --- a/docs/source/tutorials/Tutorial 02 - Loading Data from Unstructured Directory.ipynb +++ b/docs/source/tutorials/Tutorial 02 - Loading Data from Unstructured Directory.ipynb @@ -79,7 +79,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": null, "id": "4633ec79", "metadata": {}, "outputs": [ @@ -139,7 +139,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": null, "id": "adc51b1c", "metadata": {}, "outputs": [ @@ -171,7 +171,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": null, "id": "b3f8bdd9", "metadata": {}, "outputs": [], @@ -201,7 +201,7 @@ }, { "cell_type": "code", - "execution_count": 12, + "execution_count": null, "id": "25f562c0", "metadata": {}, "outputs": [ @@ -357,7 +357,7 @@ }, { "cell_type": "code", - "execution_count": 21, + "execution_count": null, "id": "5faa9122", "metadata": {}, "outputs": [], diff --git a/src/deepaudiox/modules/constructors.py b/src/deepaudiox/modules/constructors.py index 51e87cd..266bb19 100644 --- a/src/deepaudiox/modules/constructors.py +++ b/src/deepaudiox/modules/constructors.py @@ -49,7 +49,7 @@ def _resolve_backbone( downloader = Downloader() ckpt_path = downloader.download_checkpoint(backbone) ckpt = load_checkpoint(ckpt_path) - model.load_state_dict(ckpt) + model.load_state_dict(ckpt, strict=False) else: model = backbone @@ -165,7 +165,7 @@ def from_checkpoint(cls, path: str) -> "BackboneConstructor": "Cannot reconstruct model from checkpoint: a custom BaseBackbone instance was used. " "Instantiate the model manually and call model.load_state_dict(torch.load(path)['state_dict'])." ) - model = cls(**ckpt["config"]) + model = cls(**{**ckpt["config"], "pretrained": False}) model.load_state_dict(ckpt["state_dict"]) return model @@ -326,7 +326,7 @@ def from_checkpoint(cls, path: str) -> "AudioClassifierConstructor": "Cannot reconstruct model from checkpoint: a custom BaseBackbone instance was used. " "Instantiate the model manually and call model.load_state_dict(torch.load(path)['state_dict'])." ) - model = cls(**ckpt["config"]) + model = cls(**{**ckpt["config"], "pretrained": False}) model.load_state_dict(ckpt["state_dict"]) return model diff --git a/tests/test_deepaudiox.py b/tests/test_deepaudiox.py index 84797c9..b30542d 100644 --- a/tests/test_deepaudiox.py +++ b/tests/test_deepaudiox.py @@ -270,7 +270,12 @@ def test_evaluation_loop(): model = dax.AudioClassifier.from_checkpoint(str(path_to_checkpoint)) evaluator = dax.Evaluator( - test_dset=test_dataset, model=model, num_workers=4, batch_size=16, class_mapping=class_mapping, device=TEST_DEVICE + test_dset=test_dataset, + model=model, + num_workers=4, + batch_size=16, + class_mapping=class_mapping, + device=TEST_DEVICE ) evaluator.evaluate()