PytorchでMulti-GPUを試す - Qiita 贅沢な環境素晴らしい(*°∀°*) ・・・それはさておき単純にフラグを立ててパラメータを2倍にすりゃいいってもんでもないか・・・。 その前にマルチGPU環境なんてない。Google ColabのTPUで対GPUの最速に挑戦する - Qiita TP…

torch.nn.modules.activation — PyTorch master documentation def forward(self, input): return F.log_softmax(input, self.dim, _stacklevel=5) ふむふむ。結局は torch.nn.functional.log_softmax を呼ぶのか。