File size: 865 Bytes
b3f324b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from .imagebase import imagebase_ae, imagebase_ae_stride, imagebase_ae_channel
from .videobase import videobase_ae, videobase_ae_stride, videobase_ae_channel
from .videobase import (
VQVAEConfiguration,
VQVAEModel,
VQVAETrainer,
CausalVQVAEModel,
CausalVQVAEConfiguration,
CausalVQVAETrainer
)
ae_stride_config = {}
ae_stride_config.update(imagebase_ae_stride)
ae_stride_config.update(videobase_ae_stride)
ae_channel_config = {}
ae_channel_config.update(imagebase_ae_channel)
ae_channel_config.update(videobase_ae_channel)
def getae(args):
"""deprecation"""
ae = imagebase_ae.get(args.ae, None) or videobase_ae.get(args.ae, None)
assert ae is not None
return ae(args.ae)
def getae_wrapper(ae):
"""deprecation"""
ae = imagebase_ae.get(ae, None) or videobase_ae.get(ae, None)
assert ae is not None
return ae |