目录
魔改测试代码:
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 np
class 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,