模型训练错误

leoqiang10

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

求助各位大佬~


坛主liae结构的热巴丹，合成效果感觉完美，但是想自己训练的时候一直报错或者加载完没反应。。。

各种设置都自己来回开关过，bs最低开到2，但是都不行，求助各位大佬什么情况。。。。



加载样本中...: 100%|##############################################################| 1497/1497 [00:03<00:00, 468.17it/s]
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
fatal   : Memory allocation failure
Error:
Traceback (most recent call last):
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1375, in _do_call
    return fn(*args)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1360, in _run_fn
    target_list, run_metadata)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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.InternalError: 2 root error(s) found.
  (0) Internal: Attempting to perform BLAS operation using StreamExecutor without BLAS support
         [[{{node MatMul}}]]
         [[Sigmoid_2/_847]]
  (1) Internal: Attempting to perform BLAS operation using StreamExecutor without BLAS support
         [[{{node MatMul}}]]
0 successful operations.
0 derived errors ignored.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 269, in update_sample_for_preview
    self.get_history_previews()
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 389, in get_history_previews
    return self.onGetPreview (self.sample_for_preview, for_history=True)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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 "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 611, in AE_view
    self.warped_dst:warped_dst})
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 968, in run
    run_metadata_ptr)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1191, in _run
    feed_dict_tensor, options, run_metadata)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\client\session.py", line 1369, in _do_run
    run_metadata)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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.InternalError: 2 root error(s) found.
  (0) Internal: Attempting to perform BLAS operation using StreamExecutor without BLAS support
         [[node MatMul (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\Dense.py:66) ]]
         [[Sigmoid_2/_847]]
  (1) Internal: Attempting to perform BLAS operation using StreamExecutor without BLAS support
         [[node MatMul (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\Dense.py:66) ]]
0 successful operations.
0 derived errors ignored.

Errors may have originated from an input operation.
Input Source operations connected to node MatMul:
mul_6 (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\ops\__init__.py:400)
inter_AB/dense1/weight/read (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\Dense.py:47)

Input Source operations connected to node MatMul:
mul_6 (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\ops\__init__.py:400)
inter_AB/dense1/weight/read (defined at D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\Dense.py:47)

Original stack trace for 'MatMul':
  File "threading.py", line 884, in _bootstrap
  File "threading.py", line 916, in _bootstrap_inner
  File "threading.py", line 864, in run
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\mainscripts\Trainer.py", line 58, in trainerThread
    debug=debug)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 199, in __init__
    self.on_initialize()
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 416, in on_initialize
    gpu_src_inter_AB_code = self.inter_AB (gpu_src_code)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\models\ModelBase.py", line 117, in __call__
    return self.forward(*args, **kwargs)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\archis\DeepFakeArchi.py", line 151, in forward
    x = self.dense1(x)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\LayerBase.py", line 14, in __call__
    return self.forward(*args, **kwargs)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\layers\Dense.py", line 66, in forward
    x = tf.matmul(x, weight)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\util\dispatch.py", line 206, in wrapper
    return target(*args, **kwargs)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\ops\math_ops.py", line 3655, in matmul
    a, b, transpose_a=transpose_a, transpose_b=transpose_b, name=name)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\ops\gen_math_ops.py", line 5713, in mat_mul
    name=name)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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 "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\lib\site-packages\tensorflow\python\framework\ops.py", line 3569, in _create_op_internal
    op_def=op_def)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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 "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\mainscripts\Trainer.py", line 58, in trainerThread
    debug=debug)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 222, in __init__
    self.update_sample_for_preview(choose_preview_history=self.choose_preview_history)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 271, in update_sample_for_preview
    self.sample_for_preview = self.generate_next_samples()
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\models\ModelBase.py", line 467, in generate_next_samples
    sample.append ( generator.generate_next() )
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\samplelib\SampleGeneratorBase.py", line 21, in generate_next
    self.last_generation = next(self)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\samplelib\SampleGeneratorFace.py", line 112, in __next__
    return next(generator)
  File "D:\GIGA\DFL_maozhihanhua_RTX3000\_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

wtxx8888

内存分配失败，虚拟内存不够吧，虚拟内存设置方法自己百度，设100G

herrfang

看下论坛的常见错误汇总吧

奇奇怪怪1973

注意这个单词： Memory allocation failure：存储器分配失败