Skip to content
Snippets Groups Projects
Commit 74ec94a7 authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Fix interpolate

parent 69ce218f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
osrt/model.py
+ 2
− 2
View file @ 74ec94a7
......@@ -118,7 +118,7 @@ class LitSlotAttentionAutoEncoder(pl.LightningModule):
x = self.decoder_pos(x)
x = self.decoder_cnn(x.movedim(-1, 1))
x = F.interpolate(x, image.shape[-2:], mode = self.interpolate_mode)
x = F.interpolate(x, image.shape[-2:], mode='bilinear')
x = x.unflatten(0, (len(image), len(x) // len(image)))
......@@ -142,7 +142,7 @@ class LitSlotAttentionAutoEncoder(pl.LightningModule):
x = self.decoder_pos(x)
x = self.decoder_cnn(x.movedim(-1, 1))
x = F.interpolate(x, image.shape[-2:], mode = self.interpolate_mode)
x = F.interpolate(x, image.shape[-2:], mode='bilinear')
x = x.unflatten(0, (len(image), len(x) // len(image)))
......
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