Skip to content

Performance improvements and major code changes

Latest

Choose a tag to compare

@schardong schardong released this 06 Nov 11:27
· 3 commits to master since this release

Special release for SIBGRAPI 2023!
We have a tutorial session there: How to train your (neural) dragon on 2023-11-06 at 09:00 GMT-3

Major code overhaul, most significant changes are:

  • Isolated most of our code in the i3d package
  • Changed the training flow so that the data remains on the GPU as much as possible, leading to faster training and inference. Pretrained models from past releases should work out-of-the-box, they will be converted to the new format, indicated by the _v2.pth suffix in the file name. Added new pretrained models anyway, check the attached files
  • Added functions to change the w0 value. We use this to set to 1 when saving the model, thus dismissing the dependence of the experiment configuration file for inferences
  • Converted the experiment files from JSON to YAML for simplicity
  • Renamed main.py to train_sdf.py and isolated the reconstruction code in reconstruct.py.

Note that there is a ww model parameter that we were experimenting with. Setting its value independently of w0 affects training convergence dramatically in some cases, but we didn't perform major experiments on this yet. Most of the time this is set to match w0, so consider anything involving different w0 and ww values very volatile code.

What's Changed

  • Handle the U=W=0 situation in principal_directions by @DavidXu-JJ in #3

New Contributors

Full Changelog: v0.0.1-pretrained...v1.0.0