NumpyDataset now has data and targets properties
This commit is contained in:
parent
736d9a6349
commit
ee42fd68b1
@ -14,8 +14,10 @@ import torch
|
||||
|
||||
class NumpyDataset(torch.utils.data.TensorDataset):
|
||||
"""Create a PyTorch TensorDataset from NumPy arrays."""
|
||||
def __init__(self, *arrays):
|
||||
tensors = [torch.Tensor(arr) for arr in arrays]
|
||||
def __init__(self, data, targets):
|
||||
self.data = data
|
||||
self.targets = targets
|
||||
tensors = [torch.Tensor(data), torch.Tensor(targets)]
|
||||
super().__init__(*tensors)
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user