日志:
https://juejin.im/post/5bc2bd3a5188255c94465d31
lr:
optimizer用param_groups来管理参数[optimizer.param_groups],其中保存了参数组和学习率动量等,以dict的形式保存,如optimizer.param_groups['lr']
model.load_state_dict:
有两个参数,(path, strict=True)
其中strict为True则需要完全匹配,不匹配则报错;而strict为False则不需要完全匹配,若匹配到键值则赋值,否则不赋值
nelement:
torch中的nelement可以统计tensor的个数,x.nelement()
.data:
loss里的.data,对Variable类型取数值,需要用.data / .data[0]
https://blog.csdn.net/qq_27292549/article/details/80250031
用.item()代替应该也可以
squeeze、unsqueeze、cat:
https://blog.csdn.net/abc781cba/article/details/79663190
图片格式转换(tensor PIL):
https://cloud.tencent.com/developer/article/1144751
cv2读出来是bgr,PIL是rgb,要先转换;其都为HWC结构,tensor的结构为NCHW。