Source code for imaginaire.generators.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 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
[docs] def forward(self, data): r"""Dummy Generator forward. Args: data (dict): """ return