pangedeshijie

When multiple PS servers are used, How to set parameter "emb_save_dir" when using multiple PS servers？

2

I have specified the same local directory on each PS server, but the result is not good. Should this parameter be set to a public HDFS directory? Thanks!

Euler不区分有向图和无向图吗？我是无向图，数据需要double一下吗？

3

我是无向图， 只有顶点1与2相连，只有1-2 ，使用Euler训练时，需要加一个2-1吗

@alinamimi @renyi533 请教3个PS时，model_dir 是不是只能指定hdfs目录呀，我指定了本地的同名目录，结果有错误，指定hdfs目录需要配置什么呀，目录格式是什么呢？ 感谢解答！

GPU 上跑显存爆掉，有没有办法embedding在cpu上，计算在GPU上呀

1

我GPU的显存不够，有没有办法，整个embedding在cpu上，每个batch对应的embedding送到GPU上计算？

File "export.py", line 72, in y = model(img) # dry run File "E:\Anaconda\envs\yolo\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) File "F:\yolov5-face-master\models\yolo.py", line 222, in forward return self.forward_once(x, profile) #...

动态batch推理

6

你好，博主，很感谢你的项目代码，目前在用这个代码的时候遇到一点问题，我在导出onnx的时候，设置了动态batch，基于yolov5和yolov7的版本，都做了尝试，发现报了同样的错误，在多张图片同时进行推理的时候 Reshape node. Name:'/model.1/Reshape' Status Message: D:\a\_work\1\s\onnxruntime\core\providers\cpu\tensor\reshape_helper.h:41 onnxruntime::ReshapeHelper::ReshapeHelper gsl::narrow_cast(input_shape.Size()) == size was false. The input tensor cannot be reshaped to the requested shape. Input shape:{2,96,80,80}, requested shape:{1,2,48,80,80} 看着像是在某个节点reshape的时候报错了，yolov5和v7都是在同样的节点，想问下博主这种情况怎么解决？还是我操作的有问题？