1. Train reference-based colorizer

    python3 -m iconflow.train_net --output-dir output/test --device cuda

2. Train c-CNF

    python3 -m iconflow.train_flow --output-dir output/test --dirname flow --device cuda

3. Train upsampler

    python3 -m iconflow.train_up --output-dir output/test --image-size 512 --device cuda
