torch>=2.0
click>=8.0
tqdm>=4.0

[train]
numpy
tqdm
