Pytorch

PyTorchのModule#named_modulesでモデル内のすべてのOperatorにアクセスするときにはまったメモ。

PyTorch向けの軽量化ライブラリであるDistillerを触る環境をDockerで作っただけの話