code and note for creating your own NN by caffe2

最新推荐文章于 2021-05-07 13:33:33 发布

原创最新推荐文章于 2021-05-07 13:33:33 发布 · 413 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

caffe2

本文介绍如何使用Caffe2框架创建深度学习模型，包括数据准备、模型定义、权重初始化、前向传播、损失计算及模型训练过程。通过具体代码示例，展示了如何在Caffe2中实现从数据到模型训练的完整流程。

"""
Created on Sun Apr 28 10:00:14 2019

@author: wujing
"""
from caffe2.python import workspace, model_helper
import numpy as np

# =============================================================================
# x = np.random.rand(4,3,2)
# print(x)
# workspace.FeedBlob("my_x",x)
#
# x2 = workspace.FetchBlob("my_x")
# print(x2)
# =============================================================================

#create inpute data
data = np.random.rand(16,100).astype(np.float32)

#create labels for the data as integers[0,9]
label = (np.random.rand(16)*10).astype(np.int32)
workspace.FeedBlob("data",data)
workspace.FeedBlob("label",label)

'''=========define the structure and the inputs of the NN============='''
#create model using model helpler
m = model_helper.ModelHelper(name = "my fisrt net")
#generate weights and bias for FC, and store them as blobs.
weights = m.param_init_net.XavierFill([], 'fc_w', shape=[10, 100])
bias = m.param_init_net.ConstantFill([], 'fc_b', shape=[10, ])
fc_1 = m.net.FC(["data","fc_w","fc_b"],"fc_1")

#pay attention, the first param is not the string "fc_1",
#but the object fc_1. This also fit softmaxwithloss
pred = m.net.Sigmoid(fc_1,"pred")
softmax,loss = m.net.SoftmaxWithLoss([pred, "label"], ["softmax", "loss"])

'''=========execute the model=================='''
#initilize the model, by means of adding the following code,
#===>m.AddGradientOperators([loss]) as a backward function.
# it could train the model through calling .Runnet.
workspace.RunNetOnce(m.param_init_net)

#create the training model
workspace.CreateNet(m.net)

#run it multiple times.
#but This net only contains the forward pass,
#thus it is not learning anything.
for _ in range(100):
#run for 10 times, it will train the model actually after adding a backward oper
workspace.RunNet(m.name,10)

print(workspace.FetchBlob("softmax").shape)
print(workspace.FetchBlob("loss"))
#check the structure of the model
print(m.net.Proto())
#check the initial param of the model
print(m.param_init_net.Proto())