Torch.manual_seed(3407) é tudo que você precisa. Este artigo mostra que os modelos populares de aprendizado profundo oscilam no desempenho com base apenas na semente. Seu melhor modelo pode ser apenas sorte.