私人笔记
落花逐流水
从事人工智能,模式识别与智能系统
展开
-
Latex公式书写
参考:LaTeX数学公式-详细教程原创 2021-04-23 17:18:45 · 175 阅读 · 0 评论 -
pytorch转onnx支持动尺寸
def torch2onnx_dynamic(model, save_path): """ :param model: :param save_path: XXX/XXX.onnx :return: """ model.eval() data = torch.rand(1, 3, 224, 224) device = torch.device("cuda" if torch.cuda.is_available() else "cpu"...原创 2020-12-17 18:51:15 · 1113 阅读 · 0 评论