the way train mnist datasets with cnn_gcn?

[YuYue26]

when i try to train minist dataset with cnn_gcn_mnist(config), i download your data to the project file ./raw_data/MNIST.
and then, i run the python file 'python3 custom_train.py --config_dir ./configs/custom_trainer/graph_image/hybrid/cnn_gcn_mnist'.

but i occur error 'TypeError: Caught TypeError in DataLoader worker process 0.'

do i run in a right way, if not, how i can train the mnist dataset in the right way

thanks! please help me if you'd like to take a few minites!