gru

还是建议使用pytorchlightning做框架开发，速度优化什么的都没得说。 import torch import torch.nn as nn class Bigru(nn.Module): def __init__(self, num_classes, input_size, hidden_size=256, num_embeddings=21128, embedding_dim=256,...