cure-lab / MagicDrive

[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”
https://gaoruiyuan.com/magicdrive/
GNU Affero General Public License v3.0
664 stars 40 forks source link

Use of cache files #94

Open QChencq opened 1 month ago

QChencq commented 1 month ago

Hello, after I prepare the cache files in h5 format, how should I call these cache files during training?

flymin commented 1 month ago

Just follow the readme, rename the file, and put them in the default path. The code will use it if successfully locates the file. You can also change the file path through config.

QChencq commented 1 month ago

Thank you for your reply! After loading the cache file as required, I found two new logs in train.log as follows:

[2024-10-07 15:52:55,823][root][INFO] - using data cache from: MagicDrive-video/train_12Hz_advanced.h5 [2024-10-07 15:54:37,389][root][INFO] - using data cache from: MagicDrive-video/val_12Hz_advanced.h5

But I didn't find the relevant code to load the cache file. How can I check whether the cache file is loaded effectively?

flymin commented 1 month ago

You may just compare the speed between with and without cache file. The difference should be very significant.

Otherwise, you can search the line for logging, i.e., “using cache data from” . It is located in mmdet3d.

github-actions[bot] commented 1 month ago

This issue is stale because it has been open for 7 days with no activity. If you do not have any follow-ups, the issue will be closed soon.