Call ConvLSTM with gru=true

[Basharat]

class ConvLSTM(nn.Module):
def init(self, in_channels, height, width, lstm_channels=16, all_hidden=False,
mode='merge', cpt=None, dropout_rate=0.5, last_conv=False,
conv_channels=None, gru=False)

When I call ConvLSTM with gru=true the execution goes just fine for run/train BikeNYC dataset.
However test BikeNYC gives following error.

incomplete days: []
after removing 4392
Preprocessing: Min max normalizing
DataFetcher: With Length: 4, 2, 0; with Padding: 0 0, 0 0; with Interval: 1 7.
Dumped 0 data.
Traceback (most recent call last):
File "test_bikenyc.py", line 105, in
test(dconf)
File "test_bikenyc.py", line 68, in test
model = model.cuda()
AttributeError: 'collections.OrderedDict' object has no attribute 'cuda'

Any help or clue?