Fine-Tune DualStyleGAN on Target Domain auto terminate

[KkyKai]

When trying to Fine-Tune DualStyleGAN on Target Domain with

!python /content/drive/MyDrive/DualStyleGAN/finetune_dualstylegan.py --iter 1300 --batch 8 --ckpt /content/drive/MyDrive/DualStyleGAN/checkpoint/generator-pretrain.pt --style_loss 0.25 --CX_loss 0.25 --perc_loss 1 --id_loss 1 --L2_reg_loss 0.015 --augment cartoon

the code automatically terminates.

Error: load model: /content/drive/MyDrive/DualStyleGAN/checkpoint/generator-pretrain.pt Loading pSp from checkpoint: /content/drive/MyDrive/DualStyleGAN/checkpoint/encoder.pt Loading ResNet ArcFace Encoder model successfully loaded! ^C (Terminate here)

The code seems to be unable to run this lines (line 536) Simgs = torch.cat(Simgs, dim=0) # image S

Tried debugging with: for i, tensor in enumerate(Simgs): print(f"Tensor {i + 1}: Size={tensor.size()}, Dtype={tensor.dtype}") Where output is Tensor xxx: Size=torch.Size([1, 3, 1024, 1024]), Dtype=torch.float32 ^C

[KkyKai]

Solved! Seems to be not enough System Ram for concatenation of tensor with 300+ images. At least16gb system ram required