压缩自动编码器的有损图像压缩
入门
开始实验的最快方法是使用在上训练的。 通过使用提供的脚本下载框架,可以构建任意数据集。
有关更多详细信息,下载链接和更多结果,请参见 。
训练
python train.py --config ../configs/train.yaml
例如train.yaml :
exp_name : training
num_epochs : 1
batch_size : 16
learning_rate : 0.0001
# start fresh
resume : false
checkpoint : null
start_epoch : 1
batch_every : 1
save_every : 10
epoch_every : 1
shuffle : true
dataset_path : datasets/yt_small_720p
num_wo
1