B-7
commented
10 months ago hey @blingsr ! I had the same problem recently. I solved the problem by copying data/sal_datamodule.py and related files from data/ to litsr/data/.
Open blingsr opened 1 year ago
B-7
commented
10 months ago hey @blingsr ! I had the same problem recently. I solved the problem by copying data/sal_datamodule.py and related files from data/ to litsr/data/.
(SaIDRN) chenz@chenz-System-Product-Name:/media/chenz/500G/SalDRN$ python train.py --config /media/chenz/500G/SalDRN/config/sal_cssr_epsilon_fb.yaml GPU available: True, used: True TPU available: False, using: 0 TPU cores IPU available: False, using: 0 IPUs Traceback (most recent call last): File "train.py", line 106, in
train_pipeline(args)
File "train.py", line 62, in train_pipeline
datamodule = create_data_module(config.data_module)
File "/media/chenz/500G/SalDRN/litsr/data/init.py", line 31, in create_data_module
data_module = DataModuleRegistry.get(data_module_name)
File "/media/chenz/500G/SalDRN/litsr/utils/registry.py", line 67, in get
raise KeyError(
KeyError: "No object named 'SalMSDataModule' found in 'data_module' registry!"