• F
    fenglingsa

    我用torch::gt函数得到了大于阈值的index,但是这个index在libtorch里怎么用呢?没办法像python一样使用[:][:][[index]的用法啊。

    发布在 社区求助区(SOS!!) 阅读更多
  • F
    fenglingsa

    @金天 尺寸应该没问题,直接在python中训练和推测都没问题,结果也是OK的。但是一旦trace到C++就报错,而且很有趣的是,如果只是单纯生成网络,不训练,仅有自生成的随机权重时,trace到C++没有问题,如果load训练后的参数,再trace,就会出现这个错误。
    我查了一下pytorch的文档,发现官方文档里提过这个错误
    ![0_1569389525001_批注 2019-09-25 133116.jpg](正在上传 100%)
    但是,我这边不存在原地更新张量的操作,我猜测这个错误可能与pytorch内部的内存操作有关,不知道是不是我网络权重参数过多造成的。

    发布在 社区求助区(SOS!!) 阅读更多
  • F
    fenglingsa

    @金天 FPN+SSD的那个,不过代码被我做过一些修改,我把卷积层的第二层也加入了FPN结构中

    发布在 社区求助区(SOS!!) 阅读更多
  • F
    fenglingsa

    批注 2019-09-24 163202.jpg

    如果不训练,直接用随机的参数生成C++脚本,没有问题。
    把网络训练好后,再用torch.jit.trace输出成C++脚本时就会报错。

    发布在 社区求助区(SOS!!) 阅读更多

与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待