search

Henry1iu / TNT-Trajectory-Prediction

A Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
487 stars 92 forks source link

No module named 'argoverse' #61

Open SExpert12 opened 3 weeks ago

SExpert12 commented 3 weeks ago

I am running train_vectornet.py like this and I got this error.

python train_vectornet.py -- data_root /home/forecasting_sample_v1.1/forecasting_sample/data --output_dir run/tnt/ --aux_loss --batch_size 64 --with_cuda --cuda_device 0 --lr 0.0010 --warmup_epoch 30 --lr_update_freq 10 --lr_decay_rate 0.1

Traceback (most recent call last): File "train_vectornet.py", line 14, in from core.trainer.vectornet_trainer import VectorNetTrainer File "/home/TNT-Trajectory-Prediction-main/core/trainer/vectornet_trainer.py", line 9, in from argoverse.evaluation.eval_forecasting import get_displacement_errors_and_miss_rate

ModuleNotFoundError: No module named 'argoverse'

How to resolve this?