写论文需要的一些python程序小技巧

在写论文的时候经常会用到一些常用的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  

你可能感兴趣的:(python)