SystemErrorWang / White-box-Cartoonization

Official tensorflow implementation for CVPR2020 paper “Learning to Cartoonize Using White-box Cartoon Representations”
3.93k stars 737 forks source link

0 gpu train / pretrain #71

Closed hot21841 closed 3 years ago

hot21841 commented 3 years ago

(wbc) PS C:\anaconda3\White-box-Cartoonization-master\train_code> python pretrain.py C:\anaconda3\envs\wbc\lib\site-packages\numpy_distributor_init.py:32: UserWarning: loaded more than 1 DLL from .libs: C:\anaconda3\envs\wbc\lib\site-packages\numpy.libs\libopenblas.QVLO2T66WEPI7JZ63PS3HMOHFEY472BC.gfortran-win_amd64.dll C:\anaconda3\envs\wbc\lib\site-packages\numpy.libs\libopenblas.WCDJNK7YVMPZQ2ME2ZZHJJRJ3JIKNDB7.gfortran-win_amd64.dll stacklevel=1) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:523: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint8 = np.dtype([("qint8", np.int8, 1)]) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:524: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_quint8 = np.dtype([("quint8", np.uint8, 1)]) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:525: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint16 = np.dtype([("qint16", np.int16, 1)]) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:526: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_quint16 = np.dtype([("quint16", np.uint16, 1)]) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:527: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint32 = np.dtype([("qint32", np.int32, 1)]) C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\dtypes.py:532: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. np_resource = np.dtype([("resource", np.ubyte, 1)]) 2021-03-19 12:31:16.273593: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 Traceback (most recent call last): File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1334, in _do_call return fn(*args) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1317, in _run_fn self._extend_graph() File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1352, in _extend_graph tf_session.ExtendSession(self._session) tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation init: Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device. [[{{node init}} = NoOp[_device="/device:GPU:0"](^generator/Conv/weights/Assign, ^generator/Conv/biases/Assign, ^generator/Conv_1/weights/Assign, ^generator/Conv_1/biases/Assign, ^generator/Conv_2/weights/Assign, ^generator/Conv_2/biases/Assign, ^generator/Conv_3/weights/Assign, ^generator/Conv_3/biases/Assign, ^generator/Conv_4/weights/Assign, ^generator/Conv_4/biases/Assign, ^generator/block_0/conv1/weights/Assign, ^generator/block_0/conv1/biases/Assign, ^generator/block_0/conv2/weights/Assign, ^generator/block_0/conv2/biases/Assign, ^generator/block_1/conv1/weights/Assign, ^generator/block_1/conv1/biases/Assign, ^generator/block_1/conv2/weights/Assign, ^generator/block_1/conv2/biases/Assign, ^generator/block_2/conv1/weights/Assign, ^generator/block_2/conv1/biases/Assign, ^generator/block_2/conv2/weights/Assign, ^generator/block_2/conv2/biases/Assign, ^generator/block_3/conv1/weights/Assign, ^generator/block_3/conv1/biases/Assign, ^generator/block_3/conv2/weights/Assign, ^generator/block_3/conv2/biases/Assign, ^generator/Conv_5/weights/Assign, ^generator/Conv_5/biases/Assign, ^generator/Conv_6/weights/Assign, ^generator/Conv_6/biases/Assign, ^generator/Conv_7/weights/Assign, ^generator/Conv_7/biases/Assign, ^generator/Conv_8/weights/Assign, ^generator/Conv_8/biases/Assign, ^generator/Conv_9/weights/Assign, ^generator/Conv_9/biases/Assign, ^beta1_power/Assign, ^beta2_power/Assign, ^generator/Conv/weights/Adam/Assign, ^generator/Conv/weights/Adam_1/Assign, ^generator/Conv/biases/Adam/Assign, ^generator/Conv/biases/Adam_1/Assign, ^generator/Conv_1/weights/Adam/Assign, ^generator/Conv_1/weights/Adam_1/Assign, ^generator/Conv_1/biases/Adam/Assign, ^generator/Conv_1/biases/Adam_1/Assign, ^generator/Conv_2/weights/Adam/Assign, ^generator/Conv_2/weights/Adam_1/Assign, ^generator/Conv_2/biases/Adam/Assign, ^generator/Conv_2/biases/Adam_1/Assign, ^generator/Conv_3/weights/Adam/Assign, ^generator/Conv_3/weights/Adam_1/Assign, ^generator/Conv_3/biases/Adam/Assign, ^generator/Conv_3/biases/Adam_1/Assign, ^generator/Conv_4/weights/Adam/Assign, ^generator/Conv_4/weights/Adam_1/Assign, ^generator/Conv_4/biases/Adam/Assign, ^generator/Conv_4/biases/Adam_1/Assign, ^generator/block_0/conv1/weights/Adam/Assign, ^generator/block_0/conv1/weights/Adam_1/Assign, ^generator/block_0/conv1/biases/Adam/Assign, ^generator/block_0/conv1/biases/Adam_1/Assign, ^generator/block_0/conv2/weights/Adam/Assign, ^generator/block_0/conv2/weights/Adam_1/Assign, ^generator/block_0/conv2/biases/Adam/Assign, ^generator/block_0/conv2/biases/Adam_1/Assign, ^generator/block_1/conv1/weights/Adam/Assign, ^generator/block_1/conv1/weights/Adam_1/Assign, ^generator/block_1/conv1/biases/Adam/Assign, ^generator/block_1/conv1/biases/Adam_1/Assign, ^generator/block_1/conv2/weights/Adam/Assign, ^generator/block_1/conv2/weights/Adam_1/Assign, ^generator/block_1/conv2/biases/Adam/Assign, ^generator/block_1/conv2/biases/Adam_1/Assign, ^generator/block_2/conv1/weights/Adam/Assign, ^generator/block_2/conv1/weights/Adam_1/Assign, ^generator/block_2/conv1/biases/Adam/Assign, ^generator/block_2/conv1/biases/Adam_1/Assign, ^generator/block_2/conv2/weights/Adam/Assign, ^generator/block_2/conv2/weights/Adam_1/Assign, ^generator/block_2/conv2/biases/Adam/Assign, ^generator/block_2/conv2/biases/Adam_1/Assign, ^generator/block_3/conv1/weights/Adam/Assign, ^generator/block_3/conv1/weights/Adam_1/Assign, ^generator/block_3/conv1/biases/Adam/Assign, ^generator/block_3/conv1/biases/Adam_1/Assign, ^generator/block_3/conv2/weights/Adam/Assign, ^generator/block_3/conv2/weights/Adam_1/Assign, ^generator/block_3/conv2/biases/Adam/Assign, ^generator/block_3/conv2/biases/Adam_1/Assign, ^generator/Conv_5/weights/Adam/Assign, ^generator/Conv_5/weights/Adam_1/Assign, ^generator/Conv_5/biases/Adam/Assign, ^generator/Conv_5/biases/Adam_1/Assign, ^generator/Conv_6/weights/Adam/Assign, ^generator/Conv_6/weights/Adam_1/Assign, ^generator/Conv_6/biases/Adam/Assign, ^generator/Conv_6/biases/Adam_1/Assign, ^generator/Conv_7/weights/Adam/Assign, ^generator/Conv_7/weights/Adam_1/Assign, ^generator/Conv_7/biases/Adam/Assign, ^generator/Conv_7/biases/Adam_1/Assign, ^generator/Conv_8/weights/Adam/Assign, ^generator/Conv_8/weights/Adam_1/Assign, ^generator/Conv_8/biases/Adam/Assign, ^generator/Conv_8/biases/Adam_1/Assign, ^generator/Conv_9/weights/Adam/Assign, ^generator/Conv_9/weights/Adam_1/Assign, ^generator/Conv_9/biases/Adam/Assign, ^generator/Conv_9/biases/Adam_1/Assign)]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "pretrain.py", line 115, in train(args) File "pretrain.py", line 69, in train sess.run(tf.global_variables_initializer()) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 929, in run run_metadata_ptr) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1152, in _run feed_dict_tensor, options, run_metadata) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1328, in _do_run run_metadata) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\client\session.py", line 1348, in _do_call raise type(e)(node_def, op, message) tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation init: Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device. [[node init (defined at pretrain.py:69) = NoOp[_device="/device:GPU:0"](^generator/Conv/weights/Assign, ^generator/Conv/biases/Assign, ^generator/Conv_1/weights/Assign, ^generator/Conv_1/biases/Assign, ^generator/Conv_2/weights/Assign, ^generator/Conv_2/biases/Assign, ^generator/Conv_3/weights/Assign, ^generator/Conv_3/biases/Assign, ^generator/Conv_4/weights/Assign, ^generator/Conv_4/biases/Assign, ^generator/block_0/conv1/weights/Assign, ^generator/block_0/conv1/biases/Assign, ^generator/block_0/conv2/weights/Assign, ^generator/block_0/conv2/biases/Assign, ^generator/block_1/conv1/weights/Assign, ^generator/block_1/conv1/biases/Assign, ^generator/block_1/conv2/weights/Assign, ^generator/block_1/conv2/biases/Assign, ^generator/block_2/conv1/weights/Assign, ^generator/block_2/conv1/biases/Assign, ^generator/block_2/conv2/weights/Assign, ^generator/block_2/conv2/biases/Assign, ^generator/block_3/conv1/weights/Assign, ^generator/block_3/conv1/biases/Assign, ^generator/block_3/conv2/weights/Assign, ^generator/block_3/conv2/biases/Assign, ^generator/Conv_5/weights/Assign, ^generator/Conv_5/biases/Assign, ^generator/Conv_6/weights/Assign, ^generator/Conv_6/biases/Assign, ^generator/Conv_7/weights/Assign, ^generator/Conv_7/biases/Assign, ^generator/Conv_8/weights/Assign, ^generator/Conv_8/biases/Assign, ^generator/Conv_9/weights/Assign, ^generator/Conv_9/biases/Assign, ^beta1_power/Assign, ^beta2_power/Assign, ^generator/Conv/weights/Adam/Assign, ^generator/Conv/weights/Adam_1/Assign, ^generator/Conv/biases/Adam/Assign, ^generator/Conv/biases/Adam_1/Assign, ^generator/Conv_1/weights/Adam/Assign, ^generator/Conv_1/weights/Adam_1/Assign, ^generator/Conv_1/biases/Adam/Assign, ^generator/Conv_1/biases/Adam_1/Assign, ^generator/Conv_2/weights/Adam/Assign, ^generator/Conv_2/weights/Adam_1/Assign, ^generator/Conv_2/biases/Adam/Assign, ^generator/Conv_2/biases/Adam_1/Assign, ^generator/Conv_3/weights/Adam/Assign, ^generator/Conv_3/weights/Adam_1/Assign, ^generator/Conv_3/biases/Adam/Assign, ^generator/Conv_3/biases/Adam_1/Assign, ^generator/Conv_4/weights/Adam/Assign, ^generator/Conv_4/weights/Adam_1/Assign, ^generator/Conv_4/biases/Adam/Assign, ^generator/Conv_4/biases/Adam_1/Assign, ^generator/block_0/conv1/weights/Adam/Assign, ^generator/block_0/conv1/weights/Adam_1/Assign, ^generator/block_0/conv1/biases/Adam/Assign, ^generator/block_0/conv1/biases/Adam_1/Assign, ^generator/block_0/conv2/weights/Adam/Assign, ^generator/block_0/conv2/weights/Adam_1/Assign, ^generator/block_0/conv2/biases/Adam/Assign, ^generator/block_0/conv2/biases/Adam_1/Assign, ^generator/block_1/conv1/weights/Adam/Assign, ^generator/block_1/conv1/weights/Adam_1/Assign, ^generator/block_1/conv1/biases/Adam/Assign, ^generator/block_1/conv1/biases/Adam_1/Assign, ^generator/block_1/conv2/weights/Adam/Assign, ^generator/block_1/conv2/weights/Adam_1/Assign, ^generator/block_1/conv2/biases/Adam/Assign, ^generator/block_1/conv2/biases/Adam_1/Assign, ^generator/block_2/conv1/weights/Adam/Assign, ^generator/block_2/conv1/weights/Adam_1/Assign, ^generator/block_2/conv1/biases/Adam/Assign, ^generator/block_2/conv1/biases/Adam_1/Assign, ^generator/block_2/conv2/weights/Adam/Assign, ^generator/block_2/conv2/weights/Adam_1/Assign, ^generator/block_2/conv2/biases/Adam/Assign, ^generator/block_2/conv2/biases/Adam_1/Assign, ^generator/block_3/conv1/weights/Adam/Assign, ^generator/block_3/conv1/weights/Adam_1/Assign, ^generator/block_3/conv1/biases/Adam/Assign, ^generator/block_3/conv1/biases/Adam_1/Assign, ^generator/block_3/conv2/weights/Adam/Assign, ^generator/block_3/conv2/weights/Adam_1/Assign, ^generator/block_3/conv2/biases/Adam/Assign, ^generator/block_3/conv2/biases/Adam_1/Assign, ^generator/Conv_5/weights/Adam/Assign, ^generator/Conv_5/weights/Adam_1/Assign, ^generator/Conv_5/biases/Adam/Assign, ^generator/Conv_5/biases/Adam_1/Assign, ^generator/Conv_6/weights/Adam/Assign, ^generator/Conv_6/weights/Adam_1/Assign, ^generator/Conv_6/biases/Adam/Assign, ^generator/Conv_6/biases/Adam_1/Assign, ^generator/Conv_7/weights/Adam/Assign, ^generator/Conv_7/weights/Adam_1/Assign, ^generator/Conv_7/biases/Adam/Assign, ^generator/Conv_7/biases/Adam_1/Assign, ^generator/Conv_8/weights/Adam/Assign, ^generator/Conv_8/weights/Adam_1/Assign, ^generator/Conv_8/biases/Adam/Assign, ^generator/Conv_8/biases/Adam_1/Assign, ^generator/Conv_9/weights/Adam/Assign, ^generator/Conv_9/weights/Adam_1/Assign, ^generator/Conv_9/biases/Adam/Assign, ^generator/Conv_9/biases/Adam_1/Assign)]]

Caused by op 'init', defined at: File "pretrain.py", line 115, in train(args) File "pretrain.py", line 69, in train sess.run(tf.global_variables_initializer()) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\ops\variables.py", line 2854, in global_variables_initializer return variables_initializer(global_variables()) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\ops\variables.py", line 2831, in variables_initializer return control_flow_ops.group([v.initializer for v in var_list], name=name) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 3432, in group return _GroupControlDeps(dev, deps, name=name) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 3384, in _GroupControlDeps return no_op(name=name) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\ops\gen_control_flow_ops.py", line 511, in no_op "NoOp", name=name) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 787, in _apply_op_helper op_def=op_def) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\util\deprecation.py", line 488, in new_func return func(args, **kwargs) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\ops.py", line 3274, in create_op op_def=op_def) File "C:\anaconda3\envs\wbc\lib\site-packages\tensorflow\python\framework\ops.py", line 1770, in init self._traceback = tf_stack.extract_stack()

InvalidArgumentError (see above for traceback): Cannot assign a device for operation init: Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device. [[node init (defined at pretrain.py:69) = NoOp[_device="/device:GPU:0"](^generator/Conv/weights/Assign, ^generator/Conv/biases/Assign, ^generator/Conv_1/weights/Assign, ^generator/Conv_1/biases/Assign, ^generator/Conv_2/weights/Assign, ^generator/Conv_2/biases/Assign, ^generator/Conv_3/weights/Assign, ^generator/Conv_3/biases/Assign, ^generator/Conv_4/weights/Assign, ^generator/Conv_4/biases/Assign, ^generator/block_0/conv1/weights/Assign, ^generator/block_0/conv1/biases/Assign, ^generator/block_0/conv2/weights/Assign, ^generator/block_0/conv2/biases/Assign, ^generator/block_1/conv1/weights/Assign, ^generator/block_1/conv1/biases/Assign, ^generator/block_1/conv2/weights/Assign, ^generator/block_1/conv2/biases/Assign, ^generator/block_2/conv1/weights/Assign, ^generator/block_2/conv1/biases/Assign, ^generator/block_2/conv2/weights/Assign, ^generator/block_2/conv2/biases/Assign, ^generator/block_3/conv1/weights/Assign, ^generator/block_3/conv1/biases/Assign, ^generator/block_3/conv2/weights/Assign, ^generator/block_3/conv2/biases/Assign, ^generator/Conv_5/weights/Assign, ^generator/Conv_5/biases/Assign, ^generator/Conv_6/weights/Assign, ^generator/Conv_6/biases/Assign, ^generator/Conv_7/weights/Assign, ^generator/Conv_7/biases/Assign, ^generator/Conv_8/weights/Assign, ^generator/Conv_8/biases/Assign, ^generator/Conv_9/weights/Assign, ^generator/Conv_9/biases/Assign, ^beta1_power/Assign, ^beta2_power/Assign, ^generator/Conv/weights/Adam/Assign, ^generator/Conv/weights/Adam_1/Assign, ^generator/Conv/biases/Adam/Assign, ^generator/Conv/biases/Adam_1/Assign, ^generator/Conv_1/weights/Adam/Assign, ^generator/Conv_1/weights/Adam_1/Assign, ^generator/Conv_1/biases/Adam/Assign, ^generator/Conv_1/biases/Adam_1/Assign, ^generator/Conv_2/weights/Adam/Assign, ^generator/Conv_2/weights/Adam_1/Assign, ^generator/Conv_2/biases/Adam/Assign, ^generator/Conv_2/biases/Adam_1/Assign, ^generator/Conv_3/weights/Adam/Assign, ^generator/Conv_3/weights/Adam_1/Assign, ^generator/Conv_3/biases/Adam/Assign, ^generator/Conv_3/biases/Adam_1/Assign, ^generator/Conv_4/weights/Adam/Assign, ^generator/Conv_4/weights/Adam_1/Assign, ^generator/Conv_4/biases/Adam/Assign, ^generator/Conv_4/biases/Adam_1/Assign, ^generator/block_0/conv1/weights/Adam/Assign, ^generator/block_0/conv1/weights/Adam_1/Assign, ^generator/block_0/conv1/biases/Adam/Assign, ^generator/block_0/conv1/biases/Adam_1/Assign, ^generator/block_0/conv2/weights/Adam/Assign, ^generator/block_0/conv2/weights/Adam_1/Assign, ^generator/block_0/conv2/biases/Adam/Assign, ^generator/block_0/conv2/biases/Adam_1/Assign, ^generator/block_1/conv1/weights/Adam/Assign, ^generator/block_1/conv1/weights/Adam_1/Assign, ^generator/block_1/conv1/biases/Adam/Assign, ^generator/block_1/conv1/biases/Adam_1/Assign, ^generator/block_1/conv2/weights/Adam/Assign, ^generator/block_1/conv2/weights/Adam_1/Assign, ^generator/block_1/conv2/biases/Adam/Assign, ^generator/block_1/conv2/biases/Adam_1/Assign, ^generator/block_2/conv1/weights/Adam/Assign, ^generator/block_2/conv1/weights/Adam_1/Assign, ^generator/block_2/conv1/biases/Adam/Assign, ^generator/block_2/conv1/biases/Adam_1/Assign, ^generator/block_2/conv2/weights/Adam/Assign, ^generator/block_2/conv2/weights/Adam_1/Assign, ^generator/block_2/conv2/biases/Adam/Assign, ^generator/block_2/conv2/biases/Adam_1/Assign, ^generator/block_3/conv1/weights/Adam/Assign, ^generator/block_3/conv1/weights/Adam_1/Assign, ^generator/block_3/conv1/biases/Adam/Assign, ^generator/block_3/conv1/biases/Adam_1/Assign, ^generator/block_3/conv2/weights/Adam/Assign, ^generator/block_3/conv2/weights/Adam_1/Assign, ^generator/block_3/conv2/biases/Adam/Assign, ^generator/block_3/conv2/biases/Adam_1/Assign, ^generator/Conv_5/weights/Adam/Assign, ^generator/Conv_5/weights/Adam_1/Assign, ^generator/Conv_5/biases/Adam/Assign, ^generator/Conv_5/biases/Adam_1/Assign, ^generator/Conv_6/weights/Adam/Assign, ^generator/Conv_6/weights/Adam_1/Assign, ^generator/Conv_6/biases/Adam/Assign, ^generator/Conv_6/biases/Adam_1/Assign, ^generator/Conv_7/weights/Adam/Assign, ^generator/Conv_7/weights/Adam_1/Assign, ^generator/Conv_7/biases/Adam/Assign, ^generator/Conv_7/biases/Adam_1/Assign, ^generator/Conv_8/weights/Adam/Assign, ^generator/Conv_8/weights/Adam_1/Assign, ^generator/Conv_8/biases/Adam/Assign, ^generator/Conv_8/biases/Adam_1/Assign, ^generator/Conv_9/weights/Adam/Assign, ^generator/Conv_9/weights/Adam_1/Assign, ^generator/Conv_9/biases/Adam/Assign, ^generator/Conv_9/biases/Adam_1/Assign)]]

[ cuda9.0 tensorflow-gpu 1.12.0 scikit-image==0.14.5 win10-64 NVIDiA Geforce XM130 ]