imaginaire/generators/dummy.py

# Copyright (C) 2021 NVIDIA CORPORATION & AFFILIATES.  All rights reserved.
#
# This work is made available under the Nvidia Source Code License-NC.
# To view a copy of this license, check out LICENSE.md
import torch.nn as nn

from imaginaire.layers import LinearBlock


class Generator(nn.Module):
    r"""Dummy generator.

    Args:
        gen_cfg (obj): Generator definition part of the yaml config file.
        data_cfg (obj): Data definition part of the yaml config file.
    """

    def __init__(self, gen_cfg, data_cfg):
        super(Generator, self).__init__()
        self.dummy_layer = LinearBlock(1, 1)
        pass

    def forward(self, data):
        r"""Dummy Generator forward.

        Args:
            data (dict):
        """
        return