switch to internal RRDBNet with x2 support
This commit is contained in:
parent
1035915d36
commit
da6beb31b2
|
@ -5,6 +5,7 @@ from re import compile
|
|||
import torch
|
||||
from torch.onnx import export
|
||||
|
||||
from ...models.rrdb import RRDBNet
|
||||
from ...models.srvgg import SRVGGNetCompact
|
||||
from ..utils import ConversionContext, ModelDict
|
||||
|
||||
|
@ -61,8 +62,6 @@ def convert_upscale_resrgan(
|
|||
model: ModelDict,
|
||||
source: str,
|
||||
):
|
||||
from basicsr.archs.rrdbnet_arch import RRDBNet
|
||||
|
||||
name = model.get("name")
|
||||
source = source or model.get("source")
|
||||
scale = model.get("scale")
|
||||
|
|
Loading…
Reference in New Issue