Update geneformer/mtl/train.py
#582
by MajorasMeow - opened
Added changes from line 478 to 497 as proposed in /Geneformer/discussions/581 to debug max_layers_to_freeze settings being ignored.
Now user values for max_layers_to_freeze are accepted for optuna.
Thank you for your contribution!
ctheodoris changed pull request status to merged