Commit Graph

105 Commits

Author SHA1 Message Date
Quanyi Li
4affa68a06 Update doc (#20)
* Add list.py and desc

* add operations

* add structure

* update readme

* format

* update readme
2023-08-26 22:29:54 +01:00
QuanyiLi
e6831ff972 fix build error 2023-08-25 18:57:12 +01:00
Quanyi Li
e9d816ddd8 Doc-example (#18)
* reactive traffic example

* structure

* structure

* waymo example

* rename and add doc

* finish example

* example

* start from 2
2023-08-25 18:35:46 +01:00
QuanyiLi
86e0113921 doc 2023-08-08 18:18:33 +01:00
QuanyiLi
ef5621b38a readthedocs 2023-08-08 18:16:31 +01:00
QuanyiLi
d0c3c1cd4c fix path 2023-08-08 18:09:45 +01:00
QuanyiLi
3824bce8ae add requirement 2023-08-08 18:07:19 +01:00
QuanyiLi
0a5981e62c fix doc 2023-08-08 18:02:32 +01:00
QuanyiLi
52ff95a078 format 2023-08-08 17:51:04 +01:00
QuanyiLi
fac13e2e5f config readthedocs 2023-08-08 17:50:24 +01:00
Quanyi Li
64bf9811b9 Rebuttal (#15)
* pg+nuplan train

* Need map

* use gym wrapper

* use createGymWrapper

* doc

* use all scenarios!

* update 80000 scenario

* train script
2023-08-08 17:33:02 +01:00
Quanyi Li
e9c1419a91 update readme (#9) 2023-07-03 13:50:33 +01:00
Quanyi Li
5f5a5b9531 Update env (#7)
* add capture script

* gymnasium API

* training with gymnasium API
2023-06-23 19:50:40 +01:00
Quanyi Li
88b4faa00f Update video/webpage/code 2023-06-22 12:17:20 +01:00
Quanyi Li
9acdcaf321 Make video (#5)
* generate accident scene

* construction PG

* no object

* accident prob

* capture script

* update nuscenes toolds

* make video

* format

* fix test

* update readme

* update readme

* format

* format
2023-06-17 17:19:40 +01:00
Quanyi Li
db50bca7fd Add come updates for Neurips paper (#4)
* scenarionet training

* wandb

* train utils

* fix callback

* run PPO

* use pg test

* save path

* use torch

* add dependency

* update ignore

* update training

* large model

* use curriculum training

* add time to exp name

* storage_path

* restore

* update training

* use my key

* add log message

* check seed

* restore callback

* restore call bacl

* add log message

* add logging message

* restore ray1.4

* length 500

* ray 100

* wandb

* use tf

* more levels

* add callback

* 10 worker

* show level

* no env horizon

* callback result level

* more call back

* add diffuculty

* add mroen stat

* mroe stat

* show levels

* add callback

* new

* ep len 600

* fix setup

* fix stepup

* fix to 3.8

* update setup

* parallel worker!

* new exp

* add callback

* lateral dist

* pg dataset

* evaluate

* modify config

* align config

* train single RL

* update training script

* 100w eval

* less eval to reveal

* 2000 env eval

* new trianing

* eval 1000

* update eval

* more workers

* more worker

* 20 worker

* dataset to database

* split tool!

* split dataset

* try fix

* train 003

* fix mapping

* fix test

* add waymo tqdm

* utils

* fix bug

* fix bug

* waymo

* int type

* 8 worker read

* disable

* read file

* add log message

* check existence

* dist 0

* int

* check num

* suprass warning

* add filter API

* filter

* store map false

* new

* ablation

* filter

* fix

* update filyter

* reanme to from

* random select

* add overlapping checj

* fix

* new training sceheme

* new reward

* add waymo train script

* waymo different config

* copy raw data

* fix bug

* add tqdm

* update readme

* waymo

* pg

* max lateral dist 3

* pg

* crash_done instead of penalty

* no crash done

* gpu

* update eval script

* steering range penalty

* evaluate

* finish pg

* update setup

* fix bug

* test

* fix

* add on line

* train nuplan

* generate sensor

* udpate training

* static obj

* multi worker eval

* filx bug

* use ray for testing

* eval!

* filter senario

* id filter

* fox bug

* dist = 2

* filter

* eval

* eval ret

* ok

* update training pg

* test before use

* store data=False

* collect figures

* capture pic

---------

Co-authored-by: Quanyi Li <quanyi@bolei-gpu02.cs.ucla.edu>
2023-06-10 18:56:33 +01:00
QuanyiLi
41c0b01f39 update radius 2023-05-21 21:53:06 +01:00
QuanyiLi
12132bc870 remove repeat calculation 2023-05-21 14:45:01 +01:00
QuanyiLi
1c01e4417e rename scripts and update readme 2023-05-19 14:03:48 +01:00
Quanyi Li
8d0fc77aa5 Update README.md 2023-05-19 13:56:36 +01:00
Quanyi Li
00ad6f7e61 Readme (#3)
* rename to merge dataset

* add -d for operation

* test move

* add move function

* test remove

* format

* dataset -> database

* add readme

* format.sh

* test assert

* rename to database in .sh
2023-05-19 13:51:18 +01:00
Quanyi Li
37d4e72db5 Fix bug for generating dataset (#2)
* update parameters for scripts

* update write function

* modify waymo script

* use exist ok instead of overwrite

* remove TODO

* rename to comvine_dataset

* use exist_ok and force_overwrite together

* format

* test

* creat env for each thread

* restore

* fix bug

* fix pg bug

* fix

* fix bug

* add assert

* don't return done info

* to dict

* add test

* only compare sdc

* no store mao

* release memory

* add start index to argumen

* test

* format some settings/flags

* add tmp path

* add tmp dir

* test all scripts

* suppress warning

* suppress warning

* format

* test memory leak

* fix memory leak

* remove useless functions

* imap

* thread-1 process for avoiding memory leak

* add list()

* rename

* verify existence

* verify completeness

* test

* add test

* add default value

* add limit

* use script

* add anotation

* test script

* fix bug

* fix bug

* add author4

* add overwrite

* fix bug

* fix

* combine overwrite

* fix bug

* gpu007

* add result save dir

* adjust sequence

* fix test bug

* disable bash scri[t

* add episode length limit

* move scripts to root dir

* format

* fix test
2023-05-13 15:15:31 +01:00
QuanyiLi
944c7f3707 fix test 2023-05-08 17:41:22 +01:00
Quanyi Li
a8669ae730 Merge pull request #1 from metadriverse/initial-commit
Initial commit
2023-05-08 17:38:58 +01:00
QuanyiLi
cce4d2d43e add filters to combine script 2023-05-08 17:37:53 +01:00
QuanyiLi
25e911f22d fix test 2023-05-08 17:30:09 +01:00
QuanyiLi
500f69f3fb fix CI 2023-05-08 17:19:15 +01:00
QuanyiLi
e97068ca86 fix bug 2023-05-08 17:16:57 +01:00
QuanyiLi
5c3e516c06 install md for github test 2023-05-08 17:15:58 +01:00
QuanyiLi
3830fca97c remove annotation 2023-05-08 17:14:22 +01:00
QuanyiLi
f089e9e118 16 workers 2023-05-08 17:11:34 +01:00
QuanyiLi
02ddbd1662 fix bug 2023-05-08 17:09:14 +01:00
QuanyiLi
b5418ba3f6 fix bug 2023-05-08 17:07:24 +01:00
QuanyiLi
63d6cb2c1a .sh 2023-05-08 16:59:51 +01:00
QuanyiLi
8802c6dcf6 update error message 2023-05-08 16:57:36 +01:00
QuanyiLi
c080da25a7 test more scripts 2023-05-08 16:55:45 +01:00
QuanyiLi
37a7d8608e add dataset to gitignore 2023-05-08 16:31:35 +01:00
QuanyiLi
bf53687cc3 fix bug 2023-05-08 16:28:28 +01:00
QuanyiLi
3c6d562718 add num works to script arguments 2023-05-08 16:26:26 +01:00
QuanyiLi
53abbc0152 more test 2023-05-08 16:22:32 +01:00
QuanyiLi
c5b0aa8f07 fix bug 2023-05-08 16:18:20 +01:00
QuanyiLi
f610e622a8 add more test 2023-05-08 16:17:27 +01:00
QuanyiLi
348ee642c8 test combine data 2023-05-08 16:11:46 +01:00
QuanyiLi
09f46bbc3f fix combine bug 2023-05-08 16:07:05 +01:00
QuanyiLi
678cffa613 add combine API 2023-05-08 15:40:22 +01:00
QuanyiLi
fba52b7f8c batch generation 2023-05-08 15:30:51 +01:00
QuanyiLi
a9c2b58edc merge automatically 2023-05-08 15:27:22 +01:00
QuanyiLi
bbef2db107 set waymo origin path 2023-05-08 15:17:13 +01:00
QuanyiLi
2f8f9159c2 add local test for generating local dataset 2023-05-08 14:54:35 +01:00
QuanyiLi
436cbd3ab3 test waymo converter 2023-05-08 14:48:27 +01:00