babel: fix the atac loss target ordering, and trim peak memory - #1

babel: fix the atac loss target ordering, and trim peak memory#1

benjaminfreyuu merged 3 commits into

benjaminfreyuu:fix/babel-sparse-atacbenjaminfreyuu/task_predict_modality:fix/babel-sparse-atacfrom

openproblems-bio:review/babel-sparse-followupopenproblems-bio/task_predict_modality:review/babel-sparse-followupCopy head branch name to clipboard