Update auto_encoder.m