初始化模型: 100%|########################################################################| 7/7 [00:05<00:00, 1.29it/s]
实例加载中: 100%|#################################################################| 7029/7029 [00:14<00:00, 474.31it/s]
侧脸排序中: 100%|###################################################################| 128/128 [00:00<00:00, 537.06it/s]
实例加载中: 100%|#################################################################| 7910/7910 [00:16<00:00, 472.89it/s]
侧脸排序中: 100%|###################################################################| 128/128 [00:00<00:00, 425.55it/s]
fatal : Memory allocation failure
fatal : Memory allocation failure
Error:
Traceback (most recent call last):
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1375, in _do_call
return fn(*args)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1360, in _run_fn
target_list, run_metadata)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1453, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
(0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[{{node Conv2D}}]]
[[Sigmoid_2/_1183]]
(1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[{{node Conv2D}}]]
0 successful operations.
0 derived errors ignored.
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 263, in update_sample_for_preview
self.get_history_previews()
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 383, in get_history_previews
return self.onGetPreview (self.sample_for_preview, for_history=True)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 789, in onGetPreview
S, D, SS, DD, DDM, SD, SDM = [ np.clip( nn.to_data_format(x,"NHWC", self.model_data_format), 0.0, 1.0) for x in ([target_src,target_dst] + self.AE_view (target_src, target_dst) ) ]
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 611, in AE_view
self.warped_dst:warped_dst})
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 968, in run
run_metadata_ptr)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1191, in _run
feed_dict_tensor, options, run_metadata)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1369, in _do_run
run_metadata)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1394, in _do_call
raise type(e)(node_def, op, message) # pylint: disable=no-value-for-parameter
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
(0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[node Conv2D (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:101) ]]
[[Sigmoid_2/_1183]]
(1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[node Conv2D (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:101) ]]
0 successful operations.
0 derived errors ignored.
Errors may have originated from an input operation.
Input Source operations connected to node Conv2D:
Pad_4 (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:87)
encoder/down1/conv1/weight/read (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:61)
Input Source operations connected to node Conv2D:
Pad_4 (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:87)
encoder/down1/conv1/weight/read (defined at C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py:61)
Original stack trace for 'Conv2D':
File "threading.py", line 884, in _bootstrap
File "threading.py", line 916, in _bootstrap_inner
File "threading.py", line 864, in run
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 193, in __init__
self.on_initialize()
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 415, in on_initialize
gpu_src_code = self.encoder (gpu_warped_src)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\archis\DeepFakeArchi.py", line 111, in forward
x = self.down1(x)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\archis\DeepFakeArchi.py", line 43, in forward
x = self.conv1(x)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\LayerBase.py", line 14, in __call__
return self.forward(*args, **kwargs)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\leras\layers\Conv2D.py", line 101, in forward
x = tf.nn.conv2d(x, weight, strides, 'VALID', dilations=dilations, data_format=nn.data_format)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\util\dispatch.py", line 206, in wrapper
return target(*args, **kwargs)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\ops\nn_ops.py", line 2397, in conv2d
name=name)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\ops\gen_nn_ops.py", line 972, in conv2d
data_format=data_format, dilations=dilations, name=name)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 750, in _apply_op_helper
attrs=attr_protos, op_def=op_def)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\framework\ops.py", line 3569, in _create_op_internal
op_def=op_def)
File "C:\DeepFaceLab_NV30\_internal\python-3.6.8\lib\site-packages\tensorflow\python\framework\ops.py", line 2045, in __init__
self._traceback = tf_stack.extract_stack_for_node(self._c_op)
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "Trainer.py", line 46, in Trainer.trainerThread
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 216, in __init__
self.update_sample_for_preview(choose_preview_history=self.choose_preview_history)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 265, in update_sample_for_preview
self.sample_for_preview = self.generate_next_samples()
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\models\ModelBase.py", line 461, in generate_next_samples
sample.append ( generator.generate_next() )
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\samplelib\SampleGeneratorBase.py", line 21, in generate_next
self.last_generation = next(self)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\samplelib\SampleGeneratorFace.py", line 112, in __next__
return next(generator)
File "C:\DeepFaceLab_NV30\_internal\DeepFaceLab\core\joblib\SubprocessGenerator.py", line 73, in __next__
gen_data = self.cs_queue.get()
File "multiprocessing\queues.py", line 94, in get
File "multiprocessing\connection.py", line 216, in recv_bytes
File "multiprocessing\connection.py", line 318, in _recv_bytes
File "multiprocessing\connection.py", line 340, in _get_more_data
MemoryError
|Archiver|手机版|deepfacelab中文网
|网站地图
楼主
发表于 2024-3-6 14:25:47