We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9019990 commit 1095d90Copy full SHA for 1095d90
compressai_vision/codecs/sic_sfu2022.py
@@ -37,6 +37,8 @@
37
import torch
38
import torch.nn as nn
39
import torch.nn.functional as F
40
+from torch.hub import load_state_dict_from_url
41
+
42
from compressai.ans import BufferedRansEncoder, RansDecoder
43
from compressai.entropy_models import GaussianConditional
44
from compressai.layers import (
@@ -47,8 +49,6 @@
47
49
)
48
50
from compressai.models.utils import update_registered_buffers
51
from compressai.models.waseda import Cheng2020Anchor
-from torch.hub import load_state_dict_from_url
-
52
from compressai_vision.codecs.utils import crop, pad
53
from compressai_vision.registry import register_multask_codec
54
0 commit comments