ユーザー警告:RNN モジュールの重みは、メモリの単一の連続したチャンクの一部ではありません。

質問

複数の GPU を使用してモデルをトレーニングすると、次の警告がレポートされて画面を占有し、モデルのトレーニング ステータスを表示できなくなります: UserWarning:: RNN モジュールの重みは、単一の連続したメモリ チャンクの一部ではありません
。重みを圧縮するには、再度 flatten_parameters() を呼び出します。

解決

次の 3 行のコードを def forward():: の下に追加します。

        if not hasattr(self, '_flattened'):
            self.gru.flatten_parameters()
            setattr(self, '_flattened', True)