search

mlcommons / GaNDLF

A generalizable application framework for segmentation, regression, and classification using PyTorch
https://gandlf.org
Apache License 2.0
150 stars 78 forks source link

[BUG] final_preds_and_avg_probs.csv file is not generated after inference #862

Closed ujjwalbaid0408 closed 2 months ago

ujjwalbaid0408 commented 2 months ago

Describe the bug

A clear and concise description of what the bug is.

To Reproduce

Steps to reproduce the behavior:

  1. Use the data BreastMNIST from https://medmnist.com/'
  2. Run command 'python /content/GaNDLF/gandlf_run -c /content/config.yaml -i /content/medmnist/dataset/test_path.csv -m /content/model/ -t False -d cuda'

Expected behavior

After running inference for the classification task following .csv files must be generated:

But final_preds_and_avg_probs.csv and logits.csv are not generating.

Environment information