search

knowledge-computing / geolm

Repo for EMNLP23 paper "GeoLM: Empowering Language Models for Geospatially Grounded Language Understanding"
18 stars 3 forks source link

ImportError: cannot import name 'SpatialDataset' from 'dataset_loader' #2

Open mahrokh3409 opened 9 months ago

mahrokh3409 commented 9 months ago

Dear @zekun-li ,

I am attempting to run the following command for pre-training: python3 train_joint.py --model_save_dir=OUTPUT_WEIGHT_DIR --pseudo_sentence_dir='../../datasets/osm_pseudo_sent/world/' --nl_sentence_dir='../../datasets/wikidata/world_georelation/joint_v2/' --batch_size=28 --lr=1e-5 --spatial_dist_fill=900 --placename_to_osmid_path='../../datasets/osm_pseudo_sent/name-osmid-dict/placename_to_osmid.json'

However, I encountered the following error:

ImportError: cannot import name 'SpatialDataset' from 'dataset_loader'

Thank you for your assistance.

Kind Regards, Mah