训练Vanilla LSTM使用pytorch学习Levelset图像

Question

我使用称为“水平集”的有效方法生成了一系列图像。我有一个香草LSTM，我想训练它以从图像中学习Leveltset函数。我应该如何设置培训和评估渠道？ Here is the examples of the Levelset (LS) images。