# 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
[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