pytorch手動實現梯度下降法,隨機梯度法--基於logistic Regression並探索Mini batch作用

簡述 基於這次凸優化的大項目作業。 下面會圍繞着通過logistic Regression來做MNIST集上的手寫數字識別~ 以此來探索logistic Regression,梯度下降法,隨機梯度法,以及Mini batch的作用。 核心任務是實現梯度下降法和隨機梯度法。但是其他的準備工作也得做的較爲好~ 導入的包 import os import torch import torch.nn as
相關文章
相關標籤/搜索