文件名称:LotteryTicketHypothesis-TensorFlow:彩票假说文件最重要部分的实施
文件大小:198KB
文件格式:ZIP
更新时间:2024-05-29 02:35:59
machine-learning-algorithms machinelearning Python
这是Jonathan Frankle和Michael Carbin [1]在“彩票假说:训练修剪神经网络”中使用的方法的实现。 如何自己进行实验 与进行实验 python3 algorithm.py --resultdir {path to directory} 其他选项: --iterative默认为true。 定义是否应使用迭代或单次方法查找彩票。 --samplestart默认为0。以哪个样本索引开始。 如果要运行完全分离的所有样本以进行并行化,则很有用。 --samplesupremum默认为5。如上所述。 在最后一个Sample索引之上进行处理。 --reductionsteps默认为10。权重矩阵的修剪频率。 修剪多少次达到64%的重量。 --trainingsteps默认为18000。训练期间要执行多少步。 应该选择此数字,以便可以假定梯度下降已收敛。 例子:
【文件预览】:
LotteryTicketHypothesis-TensorFlow-master
----.gitignore(66B)
----ResultsIterative4Samples18000Iterations.png(97KB)
----README.md(2KB)
----plot.py(4KB)
----LICENSE(1KB)
----ResultsOneShot4Samples18000Iterations.png(105KB)
----algorithm.py(7KB)