#
# Copyright (c) Lightly AG and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
#
from dataclasses import dataclass
from typing import Iterable

from torch.nn import Module


@dataclass
class TrainableModules:
    modules: Iterable[Module]
    modules_no_weight_decay: Iterable[Module] = ()
