It seems that the currently released checkpoints ( for SegFix models) are still trained with Pytorch-0.4.1.