Hi,
Do I understand correctly that your issue is that having a checkpoint module in your model makes it non-deterministic?
But running the same code without the checkpoint, the model is deterministic?
Running your code without the checkpoint: x = self.run_func(self.conv1)(x, dummy_input) gives the same result for me.