search

Thartvigsen / StopAndHop

Code for Stop&Hop, a method for learning to classify irregularly-sampled time series early
17 stars 1 forks source link

AttributeError when loading physionet_grud_train_loader.pt #2

Open zonglunli7515 opened 2 months ago

zonglunli7515 commented 2 months ago

Hi,

When I ran the file pretrained_example.py, I encountered the following error when using torch.load to import the data:

AttributeError: Can't get attribute 'HiddenStateDataset'

Thanks in advance for your help.

Zonglun

Thartvigsen commented 1 month ago

Ah I had omitted the HiddenStateDataset class on accident. I just added it and imported it into the example script.

zonglunli7515 commented 1 month ago

Ah I had omitted the HiddenStateDataset class on accident. I just added it and imported it into the example script.

Thanks!