Infrastructure & Execution Environment

Training & Experimentation

Data Ingestion & Preprocessing

• Docker Containerization & Execution Environment
• Experiment Config (job.config.yaml)

Atlas Experiment Manager

External Tools

• NVIDIA-docker
• ffmpeg

Model Definition

• model.py

Pretrained Model

• ResNet18 (ImageNet)

Training Inference

• main.py
• train.py
• Hyperparameter Search (hparams_search.py)
• faceforensics_download.py
• download_T.sh
• restructure_data.sh
• data_loader.py

Datasets

• runs
• prepares
• loads
• input
• initializes
• triggers
• optimizes
• configures
• schedules
• supports
• executes via
• runs