- 博客(3)
- 收藏
- 关注
翻译 官方文档翻译:Tensorflow2.0中的分布式训练
概述tf.distribute.Strategy是tensorflow在多GPU、多机器或多TPU上进行分布式训练用的API.使用这个API,你能在尽可能少的改动代码的同时,分布式训练你的模型.tf.distribute.Strategy主要用于实现下面这些目标.易于使用,支持多用户段,包括研究员、机器学习工程师等 提供良好的开箱即用的性能 易于进行策略转换tf.distribu...
2020-03-22 23:32:37
1835
原创 Tensorflow2中load_weights的一些研究
假设我们定义了一个keras模型,并且使用它的save_weights函数保存了一些参数.现在我们只定义这个模型的一部分,并且使用load_weights去加载我们保存的这个完整的模型,会发生什么?首先看源代码,load_weights实际上是调用了tensorflow_core/python/keras/engine/network.py文件中Network类的load_weights函数,...
2020-02-29 00:39:58
9351
1
原创 Tensorflow2中如何处理RNN的变长输入问题
在tensorflow2中,废弃了tf.nn.dynamic_rnn函数,在tensorflow2的文档中可以看到Warning: THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Please use keras.layers.RNN(cell), ...
2020-02-28 19:57:30
1730
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人