How to load model weights in adv_scenario_gen.py?

nv-tlabs / STRIVE

Code for CVPR 2022 paper "Generating Useful Accident-Prone Driving Scenarios via a Learned Traffic Prior"

Other

111 stars 21 forks source link

How to load model weights in adv_scenario_gen.py? #3

Open gracexmatin opened 11 months ago

gracexmatin commented 11 months ago

微信截图_20231202234150

davrempe commented 11 months ago

Please make sure to carefully follow the README here to download the pre-trained model weights. You should place them in a directory called model_ckpt after downloading.

Then you can call adv_scenario_gen.py with a command that points to these weights as shown here in the README. Make sure to update the --ckpt argument to point to the downloaded weights. For example, --ckpt ./model_ckpt/traffic_model.pth.