In the current HoVerNet implementation of MONAI, it uses UpSample with NONTRAINABLE mode which used torch.nn.functional.interpolate() may cause nondeterministic.
|
self.upsample = UpSample( |
|
2, scale_factor=2, mode=UpsampleMode.NONTRAINABLE, interp_mode=InterpolateMode.BILINEAR, bias=False |
|
) |
https://pytorch.org/docs/stable/generated/torch.use_deterministic_algorithms.html#torch.use_deterministic_algorithms
Describe the solution you'd like
It may need to be explained in the docstring.
In the current
HoVerNetimplementation of MONAI, it usesUpSamplewithNONTRAINABLEmode which used torch.nn.functional.interpolate() may cause nondeterministic.MONAI/monai/networks/nets/hovernet.py
Lines 535 to 537 in d0db5fd
https://pytorch.org/docs/stable/generated/torch.use_deterministic_algorithms.html#torch.use_deterministic_algorithms
Describe the solution you'd like
It may need to be explained in the docstring.