Pytorch: 'torch.nn' n'a pas d'attribut 'TripletMarginLoss'

Créé le 10 avr. 2017  ·  3Commentaires  ·  Source: pytorch/pytorch

Appel

triplet_loss = nn.TripletMarginLoss()

résulte en:

AttributeError: module 'torch.nn' has no attribute 'TripletMarginLoss'

même avec

import torch.nn as nn

PR connexe qui a ajouté a introduit TripletMarginLoss: #1165

La solution de contournement pour le moment consiste à utiliser directement F.triplet_margin_loss , où F est import torch.nn.functional as F .

high priority

Commentaire le plus utile

Salut, la même chose se produit lorsque j'appelle m = nn.Upsample(scale_factor=2, mode='bilinear')
AttributeError : l'objet 'module' n'a pas d'attribut 'Upsample'
Mais je peux appeler avec succès nn.UpsamplingBilinear2d .
En outre, est-il prévu de mettre en œuvre le Cubic Upsample ~

Tous les 3 commentaires

Une autre solution de contournement serait d'ajouter from torch.nn.modules.loss import TripletMarginLoss .

fixé dans le maître. sera dans la prochaine version.

Salut, la même chose se produit lorsque j'appelle m = nn.Upsample(scale_factor=2, mode='bilinear')
AttributeError : l'objet 'module' n'a pas d'attribut 'Upsample'
Mais je peux appeler avec succès nn.UpsamplingBilinear2d .
En outre, est-il prévu de mettre en œuvre le Cubic Upsample ~

Cette page vous a été utile?
0 / 5 - 0 notes

Questions connexes

eliabruni picture eliabruni  ·  3Commentaires

kdexd picture kdexd  ·  3Commentaires

szagoruyko picture szagoruyko  ·  3Commentaires

soumith picture soumith  ·  3Commentaires

cdluminate picture cdluminate  ·  3Commentaires