search

deepseek-ai / DreamCraft3D

[ICLR 2024] Official implementation of DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Prior
https://mrtornado24.github.io/DreamCraft3D/
MIT License
1.96k stars 86 forks source link

Speed Up #44

Open xiaobiaodu opened 7 months ago

xiaobiaodu commented 7 months ago

Hi

I found comment the code can speed up the training time for geometry and texture stage.


trainer:
  max_steps: 5000
  log_every_n_steps: 1
  num_sanity_val_steps: 0
  val_check_interval: 200
  enable_progress_bar: true
  precision: 32
  #strategy: "ddp_find_unused_parameters_true"```
MrTornado24 commented 7 months ago

Hi, thanks for your comment. Yeah, if you are using a single GPU for training, you can comment strategy: "ddp_find_unused_parameters_true".