mirror of
https://github.com/cdtools-developers/cdtools.git
synced 2026-11-13 01:33:55 +01:00
Fix a high priority bug with the masking system which was introduced in the switch to reconstructors classes
This commit is contained in:
1 parent
fbfbf4fa87
commit
e38ebd7372
2 files changed
+5
-1
No files matched your search
@@ -190,7 +190,7 @@ class Reconstructor:
|
||||
sim_patterns = self.model.forward(*inp)
|
||||
|
||||
# Calculate the loss
|
||||
if hasattr(self, 'mask'):
|
||||
if hasattr(self.model, 'mask'):
|
||||
loss = self.model.loss(pats,
|
||||
sim_patterns,
|
||||
mask=self.model.mask)
|
||||
|
||||
Reference in new issue
Block a user