31 lines
2.2 KiB
Plaintext
31 lines
2.2 KiB
Plaintext
|
Traceback (most recent call last):
|
||
|
|
File "/home/star/whaiDir/PFCFuse/train.py", line 9, in <module>
|
||
|
|
from net import Restormer_Encoder, Restormer_Decoder, BaseFeatureExtraction, DetailFeatureExtraction
|
||
|
|
File "/home/star/whaiDir/PFCFuse/net.py", line 1, in <module>
|
||
|
|
import torch
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/__init__.py", line 2143, in <module>
|
||
|
|
from . import _meta_registrations
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_meta_registrations.py", line 9, in <module>
|
||
|
|
from torch._decomp import (
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/__init__.py", line 245, in <module>
|
||
|
|
import torch._decomp.decompositions
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/decompositions.py", line 12, in <module>
|
||
|
|
import torch._prims as prims
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 844, in <module>
|
||
|
|
ndtri = _make_elementwise_unary_prim(
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 472, in _make_elementwise_unary_prim
|
||
|
|
return _make_prim(
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 320, in _make_prim
|
||
|
|
prim_def = torch.library.custom_op(
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 142, in custom_op
|
||
|
|
return inner(fn)
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 123, in inner
|
||
|
|
result = CustomOpDef(namespace, opname, schema_str, fn)
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 169, in __init__
|
||
|
|
self._register_to_dispatcher()
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 455, in _register_to_dispatcher
|
||
|
|
lib.define(
|
||
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/library.py", line 118, in define
|
||
|
|
result = self.m.define(schema, alias_analysis, tuple(tags))
|
||
|
|
KeyboardInterrupt
|