更新kldloss,避免损失为nan Browse filesBrowse the repository at this point in the history master EgrtcommittedMay 26, 2023 1 parent75e46f1commit92139b4 Showing1 changed filewith4 additionsand3 deletions. Whitespace Ignore whitespace Split Unified...
(assuming 0...1 values) y = tf.clip_by_value(y, 1e-8, 1-1e-8) # prevent nan on log(0) rec_loss = -tf.reduce_sum(x * tf.log(y) + (1-x) * tf.log(1-y), axis=[1,2,3]) # sum the two and average over batches loss = tf.reduce_mean(kl_loss + ...