According to the latest docs, looks like you'll also need: torch.use_deterministic_algorithms(True).
確定! 回上一頁