meteorshub/relaylearning

An offline deep learning paradigm providing physical data security in multi-site medical deep learning

relaylearning's Issues

self.base_strategy has no attribute 'rgb_inputs'

Hi, thanks for your valuable work.

When I try to do the simple test as instructionhttps://github.com/MeteorsHub/RelayLearning?tab=readme-ov-file#simple-test, something went wrong in self.base_strategy, I found it is NoneType. I did not change anything in config/relay_test.yaml except for the root_path. The error is as follows:

Traceback (most recent call last):
  File "/home/lzy/mancy/RelayLearning/predict_test.py", line 223, in <module>
    predict_test()
  File "/home/lzy/mancy/RelayLearning/predict_test.py", line 82, in predict_test
    model = get_model(task_conf['model']['name'], **task_conf['model']['kwargs'])
  File "/home/lzy/mancy/RelayLearning/models/__init__.py", line 34, in get_model
    return all_models[name](**kwargs)
  File "/home/lzy/mancy/RelayLearning/models/doublegan.py", line 88, in __init__
    self.replay_post_aug_pipe = get_data_aug_pipe(rgb=self.base_strategy.rgb_inputs) if replay_post_aug else None
AttributeError: 'NoneType' object has no attribute 'rgb_inputs'