black formatter
This commit is contained in:
@@ -40,7 +40,7 @@ class GAROM(MultiSolverInterface):
|
||||
):
|
||||
"""
|
||||
Initialization of the :class:`GAROM` class.
|
||||
|
||||
|
||||
:param AbstractProblem problem: The formulation of the problem.
|
||||
:param torch.nn.Module generator: The generator model.
|
||||
:param torch.nn.Module discriminator: The discriminator model.
|
||||
@@ -50,7 +50,7 @@ class GAROM(MultiSolverInterface):
|
||||
:param Optimizer optimizer_generator: The optimizer for the generator.
|
||||
If `None`, the :class:`torch.optim.Adam` optimizer is used.
|
||||
Default is ``None``.
|
||||
:param Optimizer optimizer_discriminator: The optimizer for the
|
||||
:param Optimizer optimizer_discriminator: The optimizer for the
|
||||
discriminator. If `None`, the :class:`torch.optim.Adam` optimizer is
|
||||
used. Default is ``None``.
|
||||
:param Scheduler scheduler_generator: The learning rate scheduler for
|
||||
|
||||
Reference in New Issue
Block a user