火车PyTorch模型

PyTorch是一个Python包,提供GPU-accelerated张量计算和高水平的功能构建深度学习网络。

的MLflow PyTorch笔记本适合神经网络在MNIST手写数字识别数据和日志MLflow服务器运行结果。这日志训练在本地TensorFlow事件格式的指标和权重,然后上传MLflow运行的构件目录。最后,它开始TensorBoard并在本地读取事件记录。