We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e398eba commit cce46aeCopy full SHA for cce46ae
cellseg_models_pytorch/decoders/unet_decoder.py
@@ -103,7 +103,8 @@ def __init__(
103
enc_reductions = list(enc_reductions)
104
enc_reductions.append(1) # add the final resolution
105
up_factors = torch.tensor(enc_reductions)
106
- up_factors = (up_factors[:-1] // up_factors[1:]).tolist() # consecutive ratios
+ up_factors = (up_factors[:-1] / up_factors[1:]).tolist() # consecutive ratios
107
+ up_factors = [int(f) for f in up_factors]
108
109
# set layer-level tuple-args
110
self.long_skips = self._layer_tuple(long_skip)
0 commit comments