Pytorch中的“ tf.Session()。partial_run()”类似功能

时间:2018-09-25 01:25:14

标签: tensorflow pytorch

我正在尝试使用Pytorch重新实现项目,但是在查找Tensorflow的“ partial_run”的类似功能时遇到了一些麻烦。你能告诉我这是什么吗? 谢谢。

1 个答案:

答案 0 :(得分:1)

您不需要部分运行功能,因为Pytorch对象保留实际值(并且不只是图形中的抽象计算/输入节点,它们仅在触发计算图后才保存值-使用Session.(partial)run

因此,当您在pytorch中编写代码时,您可以(通常)将其视为编写numpy代码。同样,您在编写代码时创建的对象也将包含您可以操纵的实际值。

除了具有“传统性”之外,这还有一个很大的后果,那就是Pytorch允许您使用传统的python调试器来调试代码。