fatal : Memory allocation failure

xin376151654 · 发表于 2022-1-19 18:35:14

星级打分

1
2
3
4
5

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

Running trainer.

[new] No saved models found. Enter a name of a new model :
new

Model first run.

Choose one or several GPU idxs (separated by comma).

[CPU] : CPU
  [0] : NVIDIA GeForce RTX 3090
  [1] : NVIDIA GeForce RTX 3090
  [2] : NVIDIA GeForce RTX 3090
  [3] : NVIDIA GeForce RTX 3090
  [4] : NVIDIA GeForce RTX 3090
  [5] : NVIDIA GeForce RTX 3090
  [6] : NVIDIA GeForce RTX 3090
  [7] : NVIDIA GeForce RTX 3090

[0,1,2,3,4,5,6,7] Which GPU indexes to choose? :
0,1,2,3,4,5,6,7

[0] Autobackup every N hour ( 0..24 ?:help ) :
0
[n] Write preview history ( y/n ?:help ) :
n
[0] Target iteration :
0
[n] Flip SRC faces randomly ( y/n ?:help ) :
n
[y] Flip DST faces randomly ( y/n ?:help ) :
y
[4] Batch_size ( ?:help ) :
4
[128] Resolution ( 64-640 ?:help ) :
128
[f] Face type ( h/mf/f/wf/head ?:help ) :
f
[liae-ud] AE architecture ( ?:help ) :
liae-ud
[256] AutoEncoder dimensions ( 32-1024 ?:help ) :
256
[64] Encoder dimensions ( 16-256 ?:help ) :
64
[64] Decoder dimensions ( 16-256 ?:help ) :
64
[22] Decoder mask dimensions ( 16-256 ?:help ) :
22
[n] Eyes and mouth priority ( y/n ?:help ) :
n
[n] Uniform yaw distribution of samples ( y/n ?:help ) :
n
[n] Blur out mask ( y/n ?:help ) :
n
[y] Place models and optimizer on GPU ( y/n ?:help ) :
y
[y] Use AdaBelief optimizer? ( y/n ?:help ) :
y
[n] Use learning rate dropout ( n/y/cpu ?:help ) :
n
[y] Enable random warp of samples ( y/n ?:help ) :
y
[0.0] Random hue/saturation/light intensity ( 0.0 .. 0.3 ?:help ) :
0.0
[0.0] GAN power ( 0.0 .. 5.0 ?:help ) :
0.0
[0.0] Face style power ( 0.0..100.0 ?:help ) :
0.0
[0.0] Background style power ( 0.0..100.0 ?:help ) :
0.0
[none] Color transfer for src faceset ( none/rct/lct/mkl/idt/sot ?:help ) :
none
[n] Enable gradient clipping ( y/n ?:help ) :
n
[n] Enable pretraining mode ( y/n ?:help ) :
n
Initializing models: 100%|####################################################################################################################################################################################################| 5/5 [00:03<00:00,  1.45it/s]
Loading samples: 100%|#################################################################################################################################################################################################| 2164/2164 [00:08<00:00, 258.68it/s]
Loading samples: 100%|#################################################################################################################################################################################################| 2164/2164 [00:08<00:00, 258.19it/s]
fatal : Memory allocation failure
fatal : Memory allocation failure
请按任意键继续. . .

单显卡的时候就可以，多显卡就一直报这个错误，求大神帮忙给看看是啥问题呀