在写论文的时候经常会用到一些常用的python小技巧。在这里总结一下,方便查看。(后面想起啥我再添吧)
- 程序运行时间
import datetime
start=datetime.datetime.now() #记录下程序开始运行时的时间
# 程序段 #
end=datetime.datetime.now() #记录下程序结束时的时间
runtime=(end-start).seconds #两个时间差就是程序运行花费的时间
这个程序只是给出了程序运行需要的所有时间,跟电脑运行速度,电脑当前的状态(开了几个程序)有关,并不是真正意义上的程序运行时间。
- 使用pytorch时查看网络中间参数
查看中间参数
params=model.state_dict()
for k,v in params.items():
print(k) #打印网络中的变量名
print(params['conv1.weight']) #打印conv1的weight
print(params['conv1.bias']) #打印conv1的bias