pytorch的一些用法

日志：

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.youkuaiyun.com/qq_27292549/article/details/80250031

用.item()代替应该也可以

 

squeeze、unsqueeze、cat：

https://blog.youkuaiyun.com/abc781cba/article/details/79663190

 

图片格式转换（tensor PIL）：

https://cloud.tencent.com/developer/article/1144751

cv2读出来是bgr，PIL是rgb，要先转换；其都为HWC结构，tensor的结构为NCHW。