训练transfer learning为什么报错?
-
Traceback (most recent call last):
File "/home/train.py", line 356, in
main_function()
File "/home/train.py", line 267, in main_function
train_dataset, validation_dataset, samples_dataset, X_train, X_test, Y_train, Y_test, has_samples, X_samples, Y_samples = ei_tensorflow.training.get_dataset_from_folder(
File "/app/./resources/libraries/ei_tensorflow/training.py", line 238, in get_dataset_from_folder
X_train, X_test, Y_train, Y_test, X_train_raw = split_and_shuffle_data(
File "/app/./resources/libraries/ei_tensorflow/training.py", line 63, in split_and_shuffle_data
Y_file = np_load_file_auto_mmap(os.path.join(dir_path, y_train_path))
File "/app/./resources/libraries/ei_tensorflow/training.py", line 24, in np_load_file_auto_mmap
return np.load(file)
File "/app/keras/.venv/lib/python3.8/site-packages/numpy/lib/npyio.py", line 438, in load
raise ValueError("Cannot load file containing pickled data "
ValueError: Cannot load file containing pickled data when allow_pickle=False
Application exited with code 1
Job failed (see above)
-
参考这个代码,可以一键在colab上运行
https://github.com/SingTown/openmv_tensorflow_training_scripts/blob/main/mobilenet_v2/openmv_mobilenet_v2.ipynb