Skip to content

Commit

Permalink
Mention the retrocompatibility of changes and the need to update pred…
Browse files Browse the repository at this point in the history
…ict_config
  • Loading branch information
CharlesGaydon committed Feb 6, 2024
1 parent 9033a39 commit ca919f2
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

## 3.7.0
- Update all versions of Pytorch, Pytorch Lightning, and Pytorch Geometric.
Changes are retrocompatible for models trained with older versions (with adjustment to the configuration file).
- Refactor logging of single-class IoUs to go from num_classes+1 torchmetrics instances to only 1.

### 3.6.1
Expand Down
3 changes: 2 additions & 1 deletion run.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

TASK_NAME_DETECTION_STRING = "task.task_name="
DEFAULT_DIRECTORY = "trained_model_assets/"
DEFAULT_CONFIG_FILE = "proto151_V2.0_epoch_100_Myria3DV3.1.0_predict_config_V3.5.0.yaml"
DEFAULT_CONFIG_FILE = "proto151_V2.0_epoch_100_Myria3DV3.1.0_predict_config_V3.7.0.yaml"
DEFAULT_CHECKPOINT = "proto151_V2.0_epoch_100_Myria3DV3.1.0.ckpt"
DEFAULT_ENV = "placeholder.env"

Expand All @@ -48,6 +48,7 @@ def launch_train(
# Imports should be nested inside @hydra.main to optimize tab completion
# Read more here: https://github.com/facebookresearch/hydra/issues/934
from myria3d.train import train

utils.extras(config)

# Pretty print config using Rich library
Expand Down

0 comments on commit ca919f2

Please sign in to comment.