search

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
419 stars 22 forks source link

Some questions about model training #25

Closed 24werewolf closed 2 months ago

24werewolf commented 2 months ago

I followed the training script and the format of the training model obtained is as follows.

屏幕快照 2024-04-01 下午4 37 11

However, the format of the officially provided pre-training weights is

屏幕快照 2024-04-01 下午4 37 37
flymin commented 2 months ago

Your first figure shows the checkpoint for diffusers to resume. However, the model-to-load is saved at the end of training by default. Please configure the interval for model saving if needed.