摘要: 首先,定义你的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 阅读(68) 评论(0) 推荐(0) 编辑