Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- optimizer = tf.keras.optimizers.Adam()
- loss_object = tf.keras.losses.SparseCategoricalCrossentropy(
- from_logits=True, reduction='none')
- def loss_function(real, pred):
- mask = tf.math.logical_not(tf.math.equal(real, 0))
- loss_ = loss_object(real, pred)
- mask = tf.cast(mask, dtype=loss_.dtype)
- loss_ *= mask
- return tf.reduce_mean(loss_)
- checkpoint_dir = './training_checkpoints'
- checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
- checkpoint = tf.train.Checkpoint(optimizer=optimizer,
- encoder=encoder,
- decoder=decoder)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement