pytorch model调用时初始化