收藏本站
开启辅助访问 充值积分切换到窄版

deepfacelab中文网

 找回密码
 立即注册（仅限QQ邮箱）
搜索
deepfacelab中文网»首页 › deepfacelab专区 › 综合讨论 › 训练模型一段时间终止然后出现这个.是显存不够还是.虚拟 ...
返回列表 发新帖
查看: 874|回复: 8

训练模型一段时间终止然后出现这个.是显存不够还是.虚拟...

[复制链接]
kokia033

66

主题

219

帖子

5273

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
5273
 楼主| 发表于 2022-5-7 17:48:35 | 显示全部楼层 |阅读模式
星级打分
  • 1
  • 2
  • 3
  • 4
  • 5
 平均分:NAN  参与人数:0  我的评分:未评
Starting. Press "Enter" to stop training and save model.
Error: 2 root error(s) found.1.5001][1.0692]
  (0) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[node concat_7 (defined at E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py:1762) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

         [[gradients_1/Conv2D_20_grad/Conv2DBackpropFilter/_659]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[node concat_7 (defined at E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py:1762) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

Original stack trace for 'concat_7':
  File "threading.py", line 884, in _bootstrap
  File "threading.py", line 916, in _bootstrap_inner
  File "threading.py", line 864, in run
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\mainscripts\Trainer.py", line 58, in trainerThread
    debug=debug)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\ModelBase.py", line 193, in __init__
    self.on_initialize()
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 518, in on_initialize
    gpu_pred_src_src_d2           = self.D_src(gpu_pred_src_src_masked_opt)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\core\leras\models\ModelBase.py", line 117, in __call__
    return self.forward(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\core\leras\models\PatchDiscriminator.py", line 184, in forward
    x = tf.concat( [enc, x], axis=nn.conv2d_ch_axis)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\util\dispatch.py", line 180, in wrapper
    return target(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\array_ops.py", line 1420, in concat
    return gen_array_ops.concat_v2(values=values, axis=axis, name=name)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gen_array_ops.py", line 1464, in concat_v2
    "ConcatV2", values=values, axis=axis, name=name)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\op_def_library.py", line 794, in _apply_op_helper
    op_def=op_def)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3371, in create_op
    attrs, op_def, compute_device)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3440, in _create_op_internal
    op_def=op_def)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 1762, in __init__
    self._traceback = tf_stack.extract_stack()

Traceback (most recent call last):
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1365, in _do_call
    return fn(*args)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1350, in _run_fn
    target_list, run_metadata)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1443, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.ResourceExhaustedError: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[{{node concat_7}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

         [[gradients_1/Conv2D_20_grad/Conv2DBackpropFilter/_659]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[{{node concat_7}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\mainscripts\Trainer.py", line 129, in trainerThread
    iter, iter_time = model.train_one_iter()
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\ModelBase.py", line 474, in train_one_iter
    losses = self.onTrainOneIter()
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 774, in onTrainOneIter
    src_loss, dst_loss = self.src_dst_train (warped_src, target_src, target_srcm, target_srcm_em, warped_dst, target_dst, target_dstm, target_dstm_em)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 584, in src_dst_train
    self.target_dstm_em:target_dstm_em,
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 956, in run
    run_metadata_ptr)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1180, in _run
    feed_dict_tensor, options, run_metadata)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1359, in _do_run
    run_metadata)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\client\session.py", line 1384, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.ResourceExhaustedError: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[node concat_7 (defined at E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py:1762) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

         [[gradients_1/Conv2D_20_grad/Conv2DBackpropFilter/_659]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[3,28,256,256] and type float on /job:localhost/replica:0/task:0/deviceML:0 by allocator DmlAllocator
         [[node concat_7 (defined at E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py:1762) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

Original stack trace for 'concat_7':
  File "threading.py", line 884, in _bootstrap
  File "threading.py", line 916, in _bootstrap_inner
  File "threading.py", line 864, in run
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\mainscripts\Trainer.py", line 58, in trainerThread
    debug=debug)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\ModelBase.py", line 193, in __init__
    self.on_initialize()
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 518, in on_initialize
    gpu_pred_src_src_d2           = self.D_src(gpu_pred_src_src_masked_opt)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\core\leras\models\ModelBase.py", line 117, in __call__
    return self.forward(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\DeepFaceLab\core\leras\models\PatchDiscriminator.py", line 184, in forward
    x = tf.concat( [enc, x], axis=nn.conv2d_ch_axis)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\util\dispatch.py", line 180, in wrapper
    return target(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\array_ops.py", line 1420, in concat
    return gen_array_ops.concat_v2(values=values, axis=axis, name=name)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\ops\gen_array_ops.py", line 1464, in concat_v2
    "ConcatV2", values=values, axis=axis, name=name)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\op_def_library.py", line 794, in _apply_op_helper
    op_def=op_def)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\util\deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3371, in create_op
    attrs, op_def, compute_device)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 3440, in _create_op_internal
    op_def=op_def)
  File "E:\EDeepFaceLab_DirectX12 3000\DeepFaceLab_DirectX12\_internal\python-3.6.8\lib\site-packages\tensorflow_core\python\framework\ops.py", line 1762, in __init__
    self._traceback = tf_stack.extract_stack()

Done.
请按任意键继续. . .








回复

使用道具 举报

my117610

7

主题

136

帖子

4209

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
4209
发表于 2022-5-7 17:51:08 | 显示全部楼层
把batch_size调小，如果出现问题继续调
回复 支持 反对

使用道具 举报

daiandy

5

主题

352

帖子

4405

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
4405
发表于 2022-5-7 18:14:34 | 显示全部楼层
这个简称OOM错误，基础教程问答里有提到，
https://dfldata.xyz/forum.php?mo ... 29&extra=page=1
新人还是要多认真学习基础啊
回复 支持 反对

使用道具 举报

niubi6666

14

主题

343

帖子

3192

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
3192
发表于 2022-5-7 18:22:14 | 显示全部楼层
前面能跑是显卡努力的结果，其实它已经被塞满了。适当降低参数就好了。
回复 支持 反对

使用道具 举报

IPFS-FIL

0

主题

41

帖子

683

积分

高级丹师

Rank: 5Rank: 5

积分
683
发表于 2022-5-7 18:37:56 | 显示全部楼层
学习学习
回复

使用道具 举报

zzlxrr

3

主题

148

帖子

914

积分

高级丹师

Rank: 5Rank: 5

积分
914
发表于 2022-5-7 18:56:16 | 显示全部楼层
学习学习
回复

使用道具 举报

无绝期

12

主题

1247

帖子

7328

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
7328

万事如意节日勋章
发表于 2022-5-7 19:48:40 来自手机 | 显示全部楼层
学习学习
回复

使用道具 举报

华少

0

主题

389

帖子

4596

积分

高级丹圣

Rank: 13Rank: 13Rank: 13Rank: 13

积分
4596
发表于 2022-5-7 20:26:51 | 显示全部楼层
典型的爆显存了，关掉重新再来。
回复 支持 反对

使用道具 举报

jackyau168

21

主题

96

帖子

573

积分

高级丹师

Rank: 5Rank: 5

积分
573
发表于 2022-5-7 22:02:31 | 显示全部楼层
多數是需要下調batch size
回复 支持 反对

使用道具 举报

返回列表 发新帖

QQ|Archiver|手机版|deepfacelab中文网 |网站地图

GMT+8, 2024-11-26 22:32 , Processed in 0.152214 second(s), 38 queries .

Powered by Discuz! X3.4

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表