[TYPECHECK]
generated-members=torch.*,optuna.*,os.sched_getaffinity,iteration,metrics,preprocess
