Required packages

• pytorch-lighting
• pytorch
• pandas = 1.1.1 required by dataset.CelebA
• requests = 2.24.0
• tqdm = 4.49.0

• models.py : Define the model architeture and training, evaluation operation.
• trainer.py : Define trainer.
• nets.py : Some backbones.
• run.py : Script used for running the training and evaluation.
• utils.py : Define some functions.

In order to train or test a model, using the following script

# Specify the configuration used
config=v1.0.1.config

# Specify gpus
gpus=0

# Run
python -u run.py \
  --config config/${config} \
  --path log/${config} \
  --gpus ${gpus} \
  > log/${config}.log 2>&1