Read DAPPLE code

  1. Source
  2. tf-keras-dapple.py

Source

Github DAPPLE

commit ID: fd75dcf (v1.0.0)

tf-keras-dapple.py

L1~L25: import包, 定义flags, 打时间戳。

L102~L128: 定义命令行参数

L130: 设置HOROVOD_FUSION_THRESHOLD为0,通信聚合的阈值为0.

L132~L137: 清理log文件夹

L139~L145: 导入vgg19模型


转载请注明来源,欢迎对文章中的引用来源进行考证,欢迎指出任何有错误或不够清晰的表达。