The sampling code does not stop

wangqiang9 / SketchKnitter

About PyTorch implementation of SketchKnitter: Vectorized Sketch Generation with Diffusion Models, ICLR 2023, Spotlight.

MIT License

57 stars 9 forks source link

The sampling code does not stop #13

Closed lmagoncalo closed 11 months ago

lmagoncalo commented 11 months ago

The part for the sampling has: while len(all_images) * args.batch_size < args.num_samples: However, all_images is net updated which makes the code run infinitely.

wangqiang9 commented 11 months ago

The PR has been merged, thanks for your contribution.