2

我正在使用超参数优化包spearmint。

在代码中，我使用的是基于 keras (backend=tensorflow) 的作业。当使用 CPU 时，代码似乎运行良好。

但是，当 tensorflow 使用 GPU 时，代码终止并出现以下错误：运行作业时出现问题：

(<type 'exceptions.AttributeError'>, AttributeError("'google.protobuf.pyext._message.RepeatedScalarConta' object has no attribute '_values'",), <traceback object at 0x7fcb23ea1bd8>)
Traceback (most recent call last):
  File "/home/sujk/myLib/spearmint/spearmint/spearmint/runner.py", line 41, in job_runner
    elif job.language == PYTHON: run_python_job(job)
  File "/home/sujk/myLib/spearmint/spearmint/spearmint/runner.py", line 116, in run_python_job
    dbl_vals = param.dbl_val._values
AttributeError: 'google.protobuf.pyext._message.RepeatedScalarConta' object has no attribute '_values'
None

以下是用于重现错误的作业代码：

import numpy as np

import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout, Flatten
from keras.layers import Conv2D, MaxPooling2D
from keras import backend as K

def main(job_id, params):
    return np.random.uniform(0, 1)

if __name__ == "__main__":
    res = main(None, None)
    print(res)

以下是config.pb文件的代码：

language: PYTHON
name: "test"

variable {
    name: "k_size"
    type: INT
    size: 1
    min:  2
    max:  4
}

以上是使用以下命令编译的：

spearmint config_mnist.pb --driver=local --method=GPEIOptChooser --method-args=noiseless=0 --max-concurrent=1 --grid-size=1
4

0 回答 0