Update geneformer/mtl/train.py

#582

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

Sign up or log in to comment