Pytorch Self.network