MEGNet_PyTorch