Pytorch: 'torch.nn' não tem atributo 'TripletMarginLoss'
Chamando
triplet_loss = nn.TripletMarginLoss()
resulta em:
AttributeError: module 'torch.nn' has no attribute 'TripletMarginLoss'
mesmo com
import torch.nn as nn
PR relacionado que adicionou introduziu TripletMarginLoss: # 1165
A solução agora é usar F.triplet_margin_loss diretamente, onde F é import torch.nn.functional as F .
high priority
bartolsthoorn
Todos 3 comentários
Outra solução alternativa seria adicionar from torch.nn.modules.loss import TripletMarginLoss .
apaszke
em 10 abr. 2017
👍2
fixado no mestre. estará no próximo lançamento.
soumith
em 11 abr. 2017
🎉1
Olá, algo semelhante acontece quando ligo para m = nn.Upsample(scale_factor=2, mode='bilinear')
AttributeError: o objeto 'módulo' não tem atributo 'Upsample'
Mas posso chamar nn.UpsamplingBilinear2d sucesso.
Além disso, existe algum plano para implementar o Cubic Upsample ~
D-X-Y
em 22 jun. 2017
❤4
Esta página foi útil?
0 / 5 - 0 avaliações
Questões relacionadas
eliabruni
·
3Comentários
a1363901216
·
3Comentários
soumith
·
3Comentários
rajarshd
·
3Comentários
Coderx7
·
3Comentários
Comentários muito úteis
Olá, algo semelhante acontece quando ligo para
m = nn.Upsample(scale_factor=2, mode='bilinear')AttributeError: o objeto 'módulo' não tem atributo 'Upsample'
Mas posso chamar
nn.UpsamplingBilinear2dsucesso.Além disso, existe algum plano para implementar o
Cubic Upsample~