imaginaire/discriminators/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 Discriminator(nn.Module):
    """Dummy Discriminator constructor.

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

    def __init__(self, dis_cfg, data_cfg):
        super(Discriminator, self).__init__()
        self.dummy_layer = LinearBlock(1, 1)
        pass

    def forward(self, data):
        """Dummy discriminator forward.

        Args:
            data (dict):
        """
        return