config_prediction_AML.yaml

# Mode
mode: "Predict"

predict_file: "/efs/common_folder/240501-SCML-00697/vangalen_uniformed.val.h5ad"
batch_size: 1 # set this based on GPU memory, higher batch_size higher training speed, but also much more GPU memory will be used
model_dir: "/home/ubuntu/fulldataset/model/AML_biofoun_context/best_model/" # Path to the pretrained model
outputfile: "/home/ubuntu/fulldataset/model/AML_biofoun_context/test.Predict.csv"