Hi,
I try to use torch.use_deterministic_algorithms(True)
, but some layers in my network does not support determinism. Is there a way to set those layers that do not support determinism to work in non-deterministic mode, while the other layers in deterministic mode?
Thanks.