<reponame>jacobfulano/composer
# Copyright 2021 MosaicML. All Rights Reserved.

from dataclasses import asdict, dataclass

from composer.models.model_hparams import ModelHparams


@dataclass
class ResNet50Hparams(ModelHparams):

    def initialize_object(self):
        from composer.models import ResNet50
        return ResNet50(**asdict(self))
