Skip to content

Commit a03ea99

Browse files
Enforce eval3d with UT is set (#712)
* Enforce eval3d with UT is set * Update simple_trainer.py * format
1 parent 634cce3 commit a03ea99

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

examples/simple_trainer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1226,4 +1226,7 @@ def main(local_rank: int, world_rank, world_size: int, cfg: Config):
12261226
"and plas (via 'pip install git+https://github.com/fraunhoferhhi/PLAS.git') "
12271227
)
12281228

1229+
if cfg.with_ut:
1230+
assert cfg.with_eval3d, "Training with UT requires setting `with_eval3d` flag."
1231+
12291232
cli(main, cfg, verbose=True)

0 commit comments

Comments
 (0)