Speed up the multislice model 5x, but probably break lots of other models doing it, so remember to clean upgit status

This commit is contained in:
Abe Levitan committed 2021-01-27 22:33:45 -05:00
1 parent 0dec5aa9fd
commit 043751a8b6
5 files changed
+125 -134

No files matched your search

+1 -1
View File
@@ -121,7 +121,7 @@ class CDataset(torchdata.Dataset):
self.mask = self.mask.to(*args,**mask_kwargs)
if self.background is not None:
self.background = self.background.to(*args,**kwargs)
def get_as(self, *args, **kwargs):
"""Sets the dataset to return data on the given device and dtype