Passing dist_train instead of dist as the to_cpu paremeter to the loading functions.
Attach a file by drag & drop or click to upload
