Commit 23ffc7eb authored by Artem Pulkin's avatar Artem Pulkin

ml: earlier torch version compatibility

parent 56663873
Pipeline #44252 passed with stage
in 2 minutes and 25 seconds
......@@ -1161,7 +1161,8 @@ class Normalization:
for i in pp:
features = i.features.reshape((-1, i.features.shape[2]))
mask = i.mask.flatten()
not_mask = torch.logical_not(mask)
# not_mask = torch.logical_not(mask)
not_mask = mask == 0 # compat
features[not_mask] = pinf
mn, _ = torch.min(features, dim=0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment