Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit c9b7111

Browse files
Will FengJoelMarcey
authored andcommitted
Fix dataloader OOM in dcgan_faces_tutorial.py
1 parent 539fe31 commit c9b7111

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

beginner_source/dcgan_faces_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@
160160
dataroot = "/home/ubuntu/facebook/datasets/celeba"
161161

162162
# Number of workers for dataloader
163-
workers = 16
163+
workers = 4
164164

165165
# Batch size during training
166166
batch_size = 128

0 commit comments

Comments
 (0)