Write the model and dataset tutorials, update the dependencies, and various small bug fixes

This commit is contained in:
Abe Levitan committed 2019-10-04 16:53:22 -04:00
1 parent bc7caebdd9
commit 120e23ef15
9 files changed
+582 -6

No files matched your search

+1 -1
View File
@@ -43,7 +43,7 @@ class SimplePtycho(CDIModel):
if mask is None:
self.mask = None
else:
self.mask = t.ByteTensor(mask)
self.mask = t.BoolTensor(mask)
# We rescale the probe here so it learns at the same rate as the
# object