初始化模型
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils import gradutils
class SimpleExample(nn.Module):
def __init__(self):
super().__init__()
self.model = nn.Linear(1, 1)
def for...