tensorflow.org/tutorials/distribute/custom_training#create_the_model 说是可以 Subclassing,不过似乎目前还是 Sequential 模式比较不容易出问题。分布式训练的 debug 就比较麻烦,建议一层一层地加入然后调试是哪一层出了问题。
tensorflow.org/tutorials/distribute/custom_training#create_the_model 说是可以 Subclassing,不过似乎目前还是 Sequential 模式比较不容易出问题。分布式训练的 debug 就比较麻烦,建议一层一层地加入然后调试是哪一层出了问题。