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 32f4339 commit 4fbc02fCopy full SHA for 4fbc02f
compressai/entropy_models/entropy_models.py
@@ -54,9 +54,7 @@ def __init__(self, method):
54
55
if method not in available_entropy_coders():
56
methods = ", ".join(available_entropy_coders())
57
- raise ValueError(
58
- f'Unknown entropy coder "{method}"' f" (available: {methods})"
59
- )
+ raise ValueError(f'Unknown entropy coder "{method}" (available: {methods})')
60
61
if method == "ans":
62
from compressai import ans
@@ -474,7 +472,7 @@ def forward(
474
472
if training is None:
475
473
training = self.training
476
477
- D = x.dim()
+ D = x.dim()
478
# B C ... -> C B ...
479
perm = [1, 0] + list(range(2, D))
480
inv_perm = [0] * D
0 commit comments