diff --git a/QMLS Code/Forward-RNN.py b/QMLS Code/Forward-RNN.py index e2d9985..171c2a1 100644 --- a/QMLS Code/Forward-RNN.py +++ b/QMLS Code/Forward-RNN.py @@ -1,5 +1,5 @@ # load torch -import torch +import torch import torch.nn as nn # Define the MLMG class as a subclas of nn.Modules