lucidrains/DALLE-pytorch

Bluescreen on Windows 10

JuliusJacobsohn opened this issue · 0 comments

The error occurs in train_vae.py, somewhere in these lines.

global_step = 0
temp = STARTING_TEMP

for epoch in range(EPOCHS):
    for i, (images, _) in enumerate(distr_dl):

My setup:

  • Windows 10

  • RTX 3070 FE (8 GB)

  • 64gb RAM

  • AMD Ryzen 9 3900X (12 cores)

  • Around 30k training images across 4k folders

  • Image width: 256x256, all JPG

  • Default setup

  • Also happens when choosing 64x64 resolution via commands

The entire PC freezes with a bluescreen error and restarts automatically after a while.