![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Tensorflow
云端一散仙
主要在做图像,点云方向,欢迎讨论~
展开
-
将tensorpack的inference改为pytorch
最近在跑一个OCR模型,模型是用Tensorpack写的,模型做inference的时候,显存,速度都不是很理想,改成pytorch后,显存占用,速度比之前好了很多。记录下改inference的过程遇到的一些坑。 将pb文件转为pth文件 import torch from collections import OrderedDict import tensorflow as tf from tensorflow.python.framework import tensor_util def view_原创 2020-07-15 18:42:09 · 541 阅读 · 0 评论 -
Tensorpack初学
官方文档:Tensorpack tutorial Callbacks callbacks=[ # save the model every epoch ModelSaver(), # backup the model with best validation error MinSaver('val-error-top1'), # run inference on another Dataflow every epoch, compute classification error an原创 2020-06-19 18:26:42 · 1176 阅读 · 2 评论