Use of GPU in paysage

[Mastermind417]

Hi,

I am trying to use the GPU option as in
http://physics.bu.edu/~pankajm/ML-Notebooks/HTML/NB17_CXVI_RBM_mnist.html
I have changed the config file appropriately as in the instructions.

However, with the config file changed to "pytorch" and "gpu" I get an unexpected error:
File "rbm-train.py", line 376, in
batch_reader = batch.in_memory_batch(data[phase], batch_size, train_fraction=0.95, transform=transform)
File "paysage/paysage/batch/batch.py", line 41, in in_memory_batch
return Batch({'train': in_memory.InMemoryTable(tensor_train, batch_size, transform),
File "paysage/paysage/batch/in_memory.py", line 51, in init
self.nrows, self.ncols = be.shape(self.tensor)
File "paysage/paysage/backends/pytorch_backend/matrix.py", line 136, in shape
return tuple(tensor.size())
TypeError: 'int' object is not callable

These are the PyTorch versions and CUDA I use.
torch 1.4.0
torchvision 0.5.0
CUDA 10.1

The error still preceeds when "pytorch","cpu" are used in the backends config file.
Is there something wrong with the code written for use with pytorch?

Thank you