diff --git a/hubconf.py b/hubconf.py index fb22169..8cf3aeb 100644 --- a/hubconf.py +++ b/hubconf.py @@ -18,7 +18,7 @@ def sleepnet(pretrained=True, my_device="cpu", num_classes=2, lstm_nn_size=128, if pretrained: checkpoint = 'https://github.com/OxWearables/asleep/' \ - 'releases/download/0.0.2/bi_sleepnet.mdl' + 'releases/download/0.0.3/bi_sleepnet.mdl' model.load_state_dict( torch.hub.load_state_dict_from_url( checkpoint, diff --git a/pyproject.toml b/pyproject.toml index 80a132f..3518f77 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -17,7 +17,7 @@ name = "asleep" # Required # # For a discussion on single-sourcing the version, see # https://packaging.python.org/guides/single-sourcing-package-version/ -version = "0.2.0" # Required +version = "0.3.0" # Required # This is a one-line description or tagline of what your project does. This # corresponds to the "Summary" metadata field: