X_train_shards = np.split(X_train_permuted, args.k)
Y_train_shards = np.split(Y_train_permuted, args.k)
原因:
元素个数不能被args.k整除,导致array不能被等分
解决:
X_train, Y_train = X_train[0:806840], Y_train[0:806840]
X_train_shards = np.split(X_train_permuted, args.k)
Y_train_shards = np.split(Y_train_permuted, args.k)
原因:
元素个数不能被args.k整除,导致array不能被等分
解决:
X_train, Y_train = X_train[0:806840], Y_train[0:806840]