KeyError when initializing NicheformerFineTune – missing keys in config

[yang-lin-pp]

Hi there,
First of all, thank you for the great work on Nicheformer!
I'm currently trying to fine-tune the model using my own dataset, but I ran into a KeyError when initializing NicheformerFineTune. The error message is:
plaintext
复制
KeyError: 'baseline'
It seems that some keys expected by the NicheformerFineTune constructor are not present in the default config. For example:
'baseline'
'freeze'
'reinit_layers'
'extractor'
'regress_distribution'
'pool'
'predict_density'
'ignore_zeros'
I can work around this by using .get() with default values, but I was wondering:
Are these keys supposed to be included in the default config?
Could you provide a sample config file or a full list of required keys for fine-tuning?
Thanks in advance for your help!
Best regards