Add train_model.py for training ConvAutoencoder with fixed (256, 256) input size 8120b87 verified 24D0894 commited on Nov 2, 2024