目录
魔改测试代码:
Basisformer论文介绍
魔改测试代码:
import torch.nn as nn
import torch.nn.utils.weight_norm as wn
import matplotlib.pyplot as plt
import torch
import torch.nn.functional as F
import time
import math
import numpy as npclass MLP_bottle(nn.Module):def __init__(self, input_len, output_len, bottleneck, bias=True):super().__init__()self.linear1 = nn.Sequential(wn(nn.Linear(input_len, bottleneck, bias=bias)), nn.ReLU(), w