• Andy Craze's avatar
    Update VariationalAutoencoder.py (#5049) · 8fadb6ed
    Andy Craze authored
    Add axis=1 for to reconstruction error calculation. This will avoid taking the sum of cost so that the training objective will no longer be  batch-size dependent.
    
    Fixes #2243
    8fadb6ed
VariationalAutoencoder.py 3 KB