2024年5月21日
已经有一个预训练好的权重文件(pt文件),如何创建一个encoder。
摘要： 首先,定义你的encoder模型的架构。这个架构应该与你训练权重时使用的架构相同。你可以使用PyTorch中的nn.Module来定义模型类。 import torch import torch.nn as nn class Encoder(nn.Module): def __init__(self 阅读全文
posted @ 2024-05-21 22:57 kingchou007 阅读(55) 评论(0) 推荐(0) 编辑