steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New Dataset for per spatial cluster training

Browse Source
This commit is contained in:
Si11ium
2020-06-09 14:08:35 +02:00
parent 821b2d1961
commit 23f3aa878d
10 changed files with 104 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -25,8 +25,9 @@ def run_lightning_loop(config_obj):
# =============================================================================
# Checkpoint Saving
checkpoint_callback = ModelCheckpoint(
monitor='mean_loss',
filepath=str(logger.log_dir / 'ckpt_weights'),
verbose=True, save_top_k=0,
verbose=True, save_top_k=10,
)
# =============================================================================
@ -80,6 +81,9 @@ if __name__ == "__main__":
from _parameters import args
from ml_lib.utils.tools import fix_all_random_seeds
# When debugging, use the following parameters:
# --main_debug=True --data_worker=0
config = ThisConfig.read_namespace(args)
fix_all_random_seeds(config)
trained_model = run_lightning_loop(config)