<filename>unitorch/optim/__init__.py
# Copyright (c) FULIUCANSHENG.
# Licensed under the MIT License.

import torch
import torch.nn as nn
from torch.optim import Adam, SGD
from transformers import AdamW

<filename>unitorch/optim/__init__.py
# Copyright (c) FULIUCANSHENG.
# Licensed under the MIT License.

import torch
import torch.nn as nn
from torch.optim import Adam, SGD
from transformers import AdamW

# Copyright (c) FULIUCANSHENG. # Licensed under the MIT License.