Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GPU not working #56

Open
xiaobaiprograming opened this issue Feb 29, 2024 · 2 comments
Open

GPU not working #56

xiaobaiprograming opened this issue Feb 29, 2024 · 2 comments

Comments

@xiaobaiprograming
Copy link

OUTPUT_DIR=argoverse2.densetnt.1;
GPU_NUM=2;
python src/run.py --argoverse --argoverse2 --future_frame_num 60
--do_train --data_dir data/train/ --output_dir ${OUTPUT_DIR}
--hidden_size 128 --train_batch_size 64 --use_map
--core_num 16 --use_centerline --distributed_training ${GPU_NUM}
--other_params
semantic_lane direction l1_loss
goals_2D enhance_global_graph subdivide goal_scoring laneGCN point_sub_graph
lane_scoring complete_traj complete_traj-3 \

GPU does not work with the above command

@GentleSmile
Copy link
Member

OUTPUT_DIR=argoverse2.densetnt.1; \
GPU_NUM=2; \
python src/run.py --argoverse --argoverse2 --future_frame_num 60 \
  --do_train --data_dir data/train/ --output_dir ${OUTPUT_DIR} \
  --hidden_size 128 --train_batch_size 64 --use_map \
  --core_num 16 --use_centerline --distributed_training ${GPU_NUM} \
  --other_params \
    semantic_lane direction l1_loss \
    goals_2D enhance_global_graph subdivide goal_scoring laneGCN point_sub_graph \
    lane_scoring complete_traj complete_traj-3 \

You need append \ to the end of each line

@xiaobaiprograming
Copy link
Author

Resolved, thanks for getting back to me

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants