Keep your regular nn.Module architecture. import lightning.pytorch as pl import torch import torch.nn as nn import ... Module): def __init__(self): super().
確定! 回上一頁