thangvubk / FEQE

Official code (Tensorflow) for paper "Fast and Efficient Image Quality Enhancement via Desubpixel Convolutional Neural Networks"
MIT License
128 stars 19 forks source link

train.py #13

Open jedota opened 2 years ago

jedota commented 2 years ago

Hi! In the training section Number 2

Finetune with MSE loss on scale 4 (FEQE-P): python main.py --checkpoint checkpoint/mse_s4 --alpha_vgg 0 --pretrained_model checkpoint_test/mse_s2/model.ckpt

The main.py is not available or are you mean train.py