报错
1 cudamat.CUDAMatException: CUBLAS error
原因及处理方式:内存不够,修改shell文件内的参数设置,即调低main_mem或者gpu_mem。
2 AssertionError: Size of text_hidden2_train is not 851039
处理方式:将在目录 flickr/dbn_reps/joint_rbm_LAST下的“input_data.pbtxt"
data {
name: "text_hidden2_train"
file_pattern: "text_rbm2_LAST/train/text_hidden2-*-of-*.npy"
size: 某个数
dimensions: 1024
}
中的size修改为851039。其它类似报错可做做类似处理