set muatmul precision to highest in inference
Change the matmul precision to "highest" during inference. This guarantees deterministic behavior of the output. This is done in model.setup(stage)
Change the matmul precision to "highest" during inference. This guarantees deterministic behavior of the output. This is done in model.setup(stage)