Bluescreen on Windows 10 - Githubissues

lucidrains / DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

MIT License

5.56k stars 643 forks source link

Bluescreen on Windows 10 #420

Open JuliusJacobsohn opened 2 years ago

JuliusJacobsohn commented 2 years ago

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.