最近在学习pytorch,碰到一个问题
就是我想画出loss-weight的二维图
然后在收集loss数据时,我开始是定义了一个空的list,然后通过append方法,每次循环计算出一个loss,我就追加进去,具体代码如下:
# ================================================================== #
# 4. Training cycle #
# ================================================================== #
loss_list=[]
w_list=[]
num_epochs=100
fo