nelson1425 / EfficientAD

Unofficial implementation of EfficientAD https://arxiv.org/abs/2303.14535
https://arxiv.org/abs/2303.14535
Apache License 2.0
240 stars 62 forks source link

RuntimeError: quantile() input tensor is too large #40

Open noivan0 opened 3 months ago

noivan0 commented 3 months ago

Traceback (most recent call last): File "D:\code\git-DS\EfficientAD\efficientad.py", line 451, in main() File "D:\code\git-DS\EfficientAD\efficientad.py", line 268, in main q_st_start, q_st_end, q_ae_start, q_ae_end = map_normalization( File "C:\Users\2878045\AppData\Roaming\Python\Python39\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context return func(*args, **kwargs) File "D:\code\git-DS\EfficientAD\efficientad.py", line 374, in map_normalization q_st_start = torch.quantile(maps_st, q=0.9) RuntimeError: quantile() input tensor is too large

anyone solve this?