Skip to content

Commit

Permalink
Merge pull request #43 from Dok11/main
Browse files Browse the repository at this point in the history
sample only 4x4 tiles, since OOM when training at resolution of 1024x1024
  • Loading branch information
lucidrains authored Jan 9, 2021
2 parents d465dee + 0986c3a commit 81769ff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lightweight_gan/lightweight_gan.py
Original file line number Diff line number Diff line change
Expand Up @@ -1065,7 +1065,7 @@ def train(self):
self.steps += 1

@torch.no_grad()
def evaluate(self, num = 0, num_image_tiles = 8, trunc = 1.0):
def evaluate(self, num = 0, num_image_tiles = 4, trunc = 1.0):
self.GAN.eval()

ext = self.image_extension
Expand Down

0 comments on commit 81769ff

Please sign in to comment.