模型部署
林子要加油
从事机器学习、NLP相关工作
展开
-
docker使用GPU
参考:https://blog.csdn.net/lq1ns259ej3okyvk4jf/article/details/78764102https://blog.csdn.net/crazybean_lwb/article/details/103289614https://blog.csdn.net/qq_29300341/article/details/84754970https://...原创 2020-04-27 15:31:13 · 869 阅读 · 0 评论 -
mlflow
main.py# -*- encoding: utf-8 -*-import torchfrom torch import nnimport mlflowimport mlflow.pytorchx = torch.tensor([[1.0], [2.0], [3.0]])y = torch.tensor([[2.0], [4.0], [6.0]])class Model(n...原创 2020-04-20 21:40:31 · 281 阅读 · 0 评论