BiaPyX / BiaPy

Open source Python library for building bioimage analysis pipelines
https://BiaPyX.github.io
MIT License
114 stars 28 forks source link

Error in 3D classification notebook #58

Closed iarganda closed 5 months ago

iarganda commented 5 months ago

Running with default parameters ends up in this error when training the model:

Training configuration finished.
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
[<ipython-input-6-1fe6fcb7389d>](https://vyjefdubo0j-496ff2e9c6d22116-0-colab.googleusercontent.com/outputframe.html?vrz=colab_20240201-060115_RC00_603318229#) in <cell line: 117>()
    115 
    116 # Run the code
--> 117 biapy = BiaPy(f'/content/{job_name}.yaml', result_dir=output_path, name=job_name, run_id=1, gpu=0)
    118 biapy.run_job()

4 frames
[/usr/local/lib/python3.10/dist-packages/yacs/config.py](https://vyjefdubo0j-496ff2e9c6d22116-0-colab.googleusercontent.com/outputframe.html?vrz=colab_20240201-060115_RC00_603318229#) in _merge_a_into_b(a, b, root, key_list)
    489                 root.raise_key_rename_error(full_key)
    490             else:
--> 491                 raise KeyError("Non-existent config key: {}".format(full_key))
    492 
    493 

KeyError: 'Non-existent config key: MODEL.SPATIAL_DROPOUT'
danifranco commented 5 months ago

There is no SPATIAL_DROPOUT option, remove the following line from the notebook:

biapy_config['MODEL']['SPATIAL_DROPOUT'] =  True