文章预览 Bigru.py 2023年08月22日 640 字 还是建议使用pytorchlightning做框架开发,速度优化什么的都没得说。 ``` import torch import torch.nn as nn class Bigru(nn.Module): def __init__(self, num_classes, input_size, hidden_size=256, num_embedd… pytorch 深度学习 gru pytorchlightning 阅读全文 浏览 评论