iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update GeMOSA v4

Browse Source
This commit is contained in:
D-X-Y
2021-05-27 17:46:00 +08:00
parent b6e11c6360
commit 16861f0f3d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
xautodl/datasets/synthetic_env.py
View File

@@ -127,6 +127,10 @@ class SyntheticDEnv(data.Dataset):
targets = torch.from_numpy(targets) targets = torch.from_numpy(targets)
else: else:
targets = torch.Tensor(targets) targets = torch.Tensor(targets)
if dataset.dtype == torch.float64:
dataset = dataset.float()
if targets.dtype == torch.float64:
targets = targets.float()
return torch.Tensor([timestamp]), (dataset, targets) return torch.Tensor([timestamp]), (dataset, targets)
def __len__(self): def __len__(self):