Pytorch-Utils
Pytorch-Utils copied to clipboard
NNDataset() method
trafficstars
Thank you for sharing the repo. I am just wondering if you can share code for NNDataset() as well.
Best, Azam.
I think the NNDataset is just an abstraction of any dataset that inherits torch.utils.data.Dataset.
cannot figure out where does indices param comes from
You could use the subset class torch.utils.data.Subset(dataset, indices=train_idx).
What does 'frames' exactly mean in the code? I know it is mentioned length of sequence of the patient. But what does 'frames' mean in a generic sense ?