This commit is contained in:
Abe Levitan
2021-08-03 20:14:15 -04:00
parent 5d4fba7819
commit 24475dbdac
@@ -69,8 +69,8 @@ class PolarizedPtycho2DDataset(Ptycho2DDataset):
polarizer = []
analyzer = []
for k in range(t.tensor(translations).shape[0]):
polarizer.append((k//3)%3)
analyzer.append((k%3))
polarizer.append((k//3)%3 * 45)
analyzer.append((k%3 * 45))
self.polarizer = t.tensor(polarizer)
self.analyzer = t.tensor(analyzer)