2021
11-17
11-17
tensorflow2 自定义损失函数使用的隐藏坑
Keras的核心原则是逐步揭示复杂性,可以在保持相应的高级便利性的同时,对操作细节进行更多控制。当我们要自定义fit中的训练算法时,可以重写模型中的train_step方法,然后调用fit来训练模型。这里以tensorflow2官网中的例子来说明:importnumpyasnpimporttensorflowastffromtensorflowimportkerasx=np.random.random((1000,32))y=np.random.random((1000,1))classCustomModel(keras.Model):tf.random.set_see...
继续阅读 >