小白的博客

【代码】pytorch学习--使用m1 进行训练。

保存模型： model.save('model.h5') #model是你训练好的模型名字 加载模型： restored_model = tf.keras.models.load_model('model.h5') #加载模型

PyTorch 是一个基于 Python 的科学计算包，主要定位两类人群： NumPy 的替代品，可以利用 GPU 的性能进行计算。 深度学习研究平台拥有足够的灵活性和速度 #导入pytorch import torch import torchvision import numpy as np Tensors 类似于 NumPy 的 ndarrays ，同时 Tensors 可以使用 GPU 进行计算。 1.张量tensor a:各种数值数据称为张量 b:常数 scaler 0阶张量 c:向量 v