Source code for imaginaire.discriminators.dummy

# 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