Slide 48
Slide 48 text
47
現在のNeuron SDKのMaxPool2Dでは
4Dのinput tensor以外は未実装。
AWSに相談し、以下で対応
tensor = tensor[None,:,:,:]
tensor = self.pooling(tensor)
tensor = torch.squeeze(tensor, dim=0)
※今後、Neuron SDKのアップデートで改善予定とのフィードバックあり
モデル:質疑応答 (Question Answering)