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
import torch.nn as nn
from imaginaire.layers import LinearBlock
[docs]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
[docs] def forward(self, data): """Dummy discriminator forward. Args: data (dict): """ return