HKO-7 icon indicating copy to clipboard operation
HKO-7 copied to clipboard

a small bug in "experiments/movingmnist/mnist_rnn_main.py"

Open igloooo opened this issue 6 years ago • 1 comments

In the function "train" line 120, when creating a moving mnist factory, the parameter ctx_num is omitted, while the parameter batch_size is still divided by the number of gpus. This will cause dimension mismatch when running the file with multiple gpus.

igloooo avatar Jul 03 '18 14:07 igloooo

Yeah, thanks for pointing out this bug. I haven't tried using multiple GPUs for the MovingMNIST experiments.

sxjscience avatar Jul 06 '18 03:07 sxjscience