22 lines
628 B
Python
22 lines
628 B
Python
# Copyright (c) 2021-2025, ETH Zurich and NVIDIA CORPORATION
|
|
# All rights reserved.
|
|
#
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
"""Definitions for neural-network components for RL-agents."""
|
|
|
|
from .actor_critic import ActorCritic
|
|
from .actor_critic_recurrent import ActorCriticRecurrent
|
|
from .rnd import *
|
|
from .student_teacher import StudentTeacher
|
|
from .student_teacher_recurrent import StudentTeacherRecurrent
|
|
from .symmetry import *
|
|
from .discriminator_multi import DiscriminatorMulti
|
|
__all__ = [
|
|
"ActorCritic",
|
|
"ActorCriticRecurrent",
|
|
"StudentTeacher",
|
|
"StudentTeacherRecurrent",
|
|
"DiscriminatorMulti",
|
|
]
|