登录 注册
当前位置:主页 > 资源下载 > 49 > TensorFlow中LotteryTicketHypothesis的实现:彩票假说文件的关键部分

TensorFlow中LotteryTicketHypothesis的实现:彩票假说文件的关键部分

  • 更新:2024-07-29 16:34:56
  • 大小:198KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

这是Jonathan Frankle和Michael Carbin [1]在“彩票假说:训练修剪神经网络”中使用的方法的实现。 如何自己进行实验 与进行实验 python3 algorithm.py --resultdir {path to directory} 其他选项: --iterative默认为true。 定义是否应使用迭代或单次方法查找彩票。 --samplestart默认为0。以哪个样本索引开始。 如果要运行完全分离的所有样本以进行并行化,则很有用。 --samplesupremum默认为5。如上所述。 在最后一个Sample索引之上进行处理。 --reductionsteps默认为10。权重矩阵的修剪频率。 修剪多少次达到64%的重量。 --trainingsteps默认为18000。训练期间要执行多少步。 应该选择此数字,以便可以假定梯度下降已收敛。 例子: