Pytorch: 「torch.nn」には属性「TripletMarginLoss」がありません

作成日 2017年04月10日  ·  3コメント  ·  ソース: pytorch/pytorch

呼び出し

triplet_loss = nn.TripletMarginLoss()

結果:

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

でもで

import torch.nn as nn

導入されたTripletMarginLossを追加した関連PR:#1165

今のところ回避策はF.triplet_margin_loss直接使用することです。ここで、 Fimport torch.nn.functional as Fです。

high priority

最も参考になるコメント

こんにちは、 m = nn.Upsample(scale_factor=2, mode='bilinear')に電話すると同様のことが起こります
AttributeError: 'モジュール'オブジェクトに属性 'アップサンプル'がありません
しかし、 nn.UpsamplingBilinear2dを正常に呼び出すことができます。
その上、 Cubic Upsampleを実装する計画はありますか?

全てのコメント3件

別の回避策は、 from torch.nn.modules.loss import TripletMarginLossを追加することです。

マスターで修正されました。 次のリリースになります。

こんにちは、 m = nn.Upsample(scale_factor=2, mode='bilinear')に電話すると同様のことが起こります
AttributeError: 'モジュール'オブジェクトに属性 'アップサンプル'がありません
しかし、 nn.UpsamplingBilinear2dを正常に呼び出すことができます。
その上、 Cubic Upsampleを実装する計画はありますか?

このページは役に立ちましたか?
0 / 5 - 0 評価