numpy
torch
tensorboard
tqdm
matplotlib
hydra-core
omegaconf
wandb
dotmap
opencv-python