Closed AndreaPi closed 1 year ago
The PyTorch Lightning example uses a DCRNN, but I want to use a GConvLSTM. However, this layer doesn't return a single tensor, but three. How should I modify the training_step and the validation_step in order to be able to use a GConvLSTM?
training_step
validation_step
The PyTorch Lightning example uses a DCRNN, but I want to use a GConvLSTM. However, this layer doesn't return a single tensor, but three. How should I modify the
training_step
and thevalidation_step
in order to be able to use a GConvLSTM?