並前往https://discuss.pytorch.org/t/calculate-the-average-model-for-kfold-cross-validation-models/123321
