yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
https://arxiv.org/abs/2011.13456
Apache License 2.0
1.58k stars 295 forks source link

Segment fault #33

Open Susan19900316 opened 1 year ago

Susan19900316 commented 1 year ago

Thank your for your work.

I use python3.8, pytorch=1.7.1 or 1.7.0. always segment fault.

mo666666 commented 1 year ago

Same with my situation. Do you solve this issue?

lujiazho commented 6 months ago

Mine is below and it works

Python 3.8.1
torch                         2.0.1
torchsummary                  1.5.1
torchvision                   0.15.2