我在jupyter笔记本上运行sketch_rnn.ipynb时,在加载环境以加载训练后的数据集时,返回错误“ allow_pickle = False时无法加载对象数组”
这是Google开发人员在开发sketch_rnn算法(甚至在google colab中运行)时已经使用的代码。过去我自己在google colab上运行过它,但是它似乎无法在我自己的jupyter笔记本上运行
from magenta.models.sketch_rnn.sketch_rnn_train import *
from magenta.models.sketch_rnn.model import *
from magenta.models.sketch_rnn.utils import *
from magenta.models.sketch_rnn.rnn import *
model_params.batch_size = 1
eval_model_params = sketch_rnn_model.copy_hparams(model_params)
eval_model_params.use_input_dropout = 0
eval_model_params.use_recurrent_dropout = 0
eval_model_params.use_output_dropout = 0
eval_model_params.is_training = 0
sample_model_params = sketch_rnn_model.copy_hparams(eval_model_params)
sample_model_params.max_seq_len = 1
return [model_params, eval_model_params, sample_model_params]
[train_set, valid_set, test_set, hps_model, eval_hps_model,
sample_hps_model] = load_env_compatible(data_dir, model_dir)
Run Code Online (Sandbox Code Playgroud)
我期望输出是
INFO:tensorflow:Downloading http://github.com/hardmaru/sketch-rnn-
datasets/raw/master/aaron_sheep/aaron_sheep.npz
INFO:tensorflow:Loaded 7400/300/300 from aaron_sheep.npz
INFO:tensorflow:Dataset combined: 8000 (7400/300/300), avg len 125
INFO:tensorflow:model_params.max_seq_len 250.
total images <= …Run Code Online (Sandbox Code Playgroud) 当我加载“路透社”数据集时,出现错误
我尝试使用一些:
np_load_old = np.load
np.load = lambda *a,**k: np_load_old(*a, allow_pickle=True, **k)
(train_data, train_labels), (test_data, test_labels) = reuters.load_data(num_words=10000)
np.load = np_load_old
Run Code Online (Sandbox Code Playgroud)
原来是:
from keras.datasets import reuters
import numpy as np
(train_data, train_labels), (test_data, test_labels) = reuters.load_data(num_words=10000)
Run Code Online (Sandbox Code Playgroud)
TypeError: () 为关键字参数“allow_pickle”获得了多个值
完整追溯:
TypeError Traceback (most recent call last)
<ipython-input-11-8669b9ae66ea> in <module>()
1 from keras.datasets import reuters
----> 2 (train_data, train_labels), (test_data, test_labels) = reuters.load_data(num_words=10000)
3
4
2 frames
<ipython-input-2-8333ca7e6c7f> in <lambda>(*a, **k)
4
5 np_load_old = np.load
----> 6 np.load = lambda …Run Code Online (Sandbox Code Playgroud)