• Edward Li's avatar
    Fix `AutoencoderTiny` with `use_slicing` (#6850) · 493228a7
    Edward Li authored
    * Fix `AutoencoderTiny` with `use_slicing`
    
    When using slicing with AutoencoderTiny, the encoder mistakenly encodes the entire batch for every image in the batch.
    
    * Fixed formatting issue
    493228a7
autoencoder_tiny.py 15.6 KB