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

Training running

Browse Source
This commit is contained in:
Si11ium 2020-06-08 10:23:12 +02:00
parent 27ae8467fc
commit 821b2d1961
3 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
_parameters.py
View File

@ -31,7 +31,7 @@ main_arg_parser.add_argument("--data_use_preprocessed", type=strtobool, default=
# Training # Training
main_arg_parser.add_argument("--train_outpath", type=str, default="output", help="") main_arg_parser.add_argument("--train_outpath", type=str, default="output", help="")
main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="") main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="")
main_arg_parser.add_argument("--train_epochs", type=int, default=500, help="") main_arg_parser.add_argument("--train_epochs", type=int, default=250, help="")
main_arg_parser.add_argument("--train_batch_size", type=int, default=10, help="") main_arg_parser.add_argument("--train_batch_size", type=int, default=10, help="")
main_arg_parser.add_argument("--train_lr", type=float, default=1e-3, help="") main_arg_parser.add_argument("--train_lr", type=float, default=1e-3, help="")
main_arg_parser.add_argument("--train_weight_decay", type=float, default=1e-8, help="") main_arg_parser.add_argument("--train_weight_decay", type=float, default=1e-8, help="")

15
multi_run.py
View File

@ -1,26 +1,27 @@
import warnings import warnings
from _templates.new_project.utils.project_config import Config
warnings.filterwarnings('ignore', category=FutureWarning) warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning) warnings.filterwarnings('ignore', category=UserWarning)
# Imports # Imports
# ============================================================================= # =============================================================================
from _templates.new_project.main import run_lightning_loop, args from main import run_lightning_loop
from utils.project_config import ThisConfig
from _parameters import args
if __name__ == '__main__': if __name__ == '__main__':
# Model Settings # Model Settings
config = Config().read_namespace(args) config = ThisConfig().read_namespace(args)
# bias, activation, model, norm, max_epochs # bias, activation, model, norm, max_epochs
cnn_classifier = dict(train_epochs=10, model_use_bias=True, model_use_norm=True, data_batchsize=512) pn2 = dict(model_type='PN2',model_use_bias=True, model_use_norm=True, data_batchsize=250)
p2g = dict(model_type='P2G', model_use_bias=True, model_use_norm=True, data_batchsize=250)
# bias, activation, model, norm, max_epochs # bias, activation, model, norm, max_epochs
for arg_dict in [cnn_classifier]: for arg_dict in [p2g]:
for seed in range(5): for seed in range(10):
arg_dict.update(main_seed=seed) arg_dict.update(main_seed=seed)
config = config.update(arg_dict) config = config.update(arg_dict)

2
utils/module_mixins.py
View File

@ -213,6 +213,8 @@ class BaseValMixin:
disp.plot(include_values=True) disp.plot(include_values=True)
self.logger.log_image('Confusion Matrix', image=disp.figure_, step=self.current_epoch) self.logger.log_image('Confusion Matrix', image=disp.figure_, step=self.current_epoch)
plt.close('all')
return summary_dict return summary_dict