references/classification/utils.py
references/optical_flow/utils.py

references/classification/train.py
references/optical_flow/train.py