Hi everyone ,
I successfully trained the model and merged the LoRA weights with the base model. However, when running the evaluation script, I encounter the following error:
File "/home/xxx/segearth_r2/eval/eval.py", line 24, in
from segearth_r2.datasets.dataset import DataCollatorForCOCODatasetV2, LaSeRSDataset, RRSISDDataset, RefSegRSDataset
ImportError: cannot import name 'RRSISDDataset' from 'segearth_r2.datasets.dataset' (/home/xxx/segearth_r2/datasets/dataset.py)
running module:
deepspeed --include localhost:0 --master_port=29500 segearth_r2/eval/eval.py
...
Hi everyone ,
I successfully trained the model and merged the LoRA weights with the base model. However, when running the evaluation script, I encounter the following error:
File "/home/xxx/segearth_r2/eval/eval.py", line 24, in
from segearth_r2.datasets.dataset import DataCollatorForCOCODatasetV2, LaSeRSDataset, RRSISDDataset, RefSegRSDataset
ImportError: cannot import name 'RRSISDDataset' from 'segearth_r2.datasets.dataset' (/home/xxx/segearth_r2/datasets/dataset.py)
running module:
deepspeed --include localhost:0 --master_port=29500 segearth_r2/eval/eval.py
...