if __name__ == '__main__':
args = _parse_args()
gin_bindings = args.config
if args.mrunner:
from alpacka.utils import mrunner_client # Lazy import
spec_path = gin_bindings.pop()
specification, overrides = mrunner_client.get_configuration(spec_path)
gin_bindings.extend(overrides)
try:
neptune_logger = mrunner_client.configure_neptune(specification)
metric_logging.register_logger(neptune_logger)
except KeyError:
print('HINT: To run with Neptune logging please set your '
'NEPTUNE_API_TOKEN environment variable')
if args.tensorboard:
from alpacka.utils import tensorboard # Lazy import
tensorboard_logger = tensorboard.TensorBoardLogger(args.output_dir)
metric_logging.register_logger(tensorboard_logger)
gin.parse_config_files_and_bindings(args.config_file, gin_bindings)
runner = Runner(args.output_dir)
runner.run()
def connect_to_neptune_experiment_add_logger(project_id, experiment_id):
neptune.init(project_id)
exp = neptune.project.get_experiments(id=experiment_id)[0]
metric_logging.register_logger(NeptuneLogger(exp))
action='store_true',
help='Add mrunner spec to gin-config overrides and Neptune to loggers.'
'\nNOTE: It assumes that the last config override (--config argument) '
'is a path to a pickled experiment config created by the mrunner CLI or'
'a mrunner specification file.')
return parser.parse_args()
if __name__ == '__main__':
args = _parse_args()
gin_bindings = args.config
if args.mrunner:
from alpacka.utils import mrunner_client # Lazy import
spec_path = gin_bindings.pop()
specification, overrides = mrunner_client.get_configuration(spec_path)
gin_bindings.extend(overrides)
try:
neptune_logger = mrunner_client.configure_neptune(specification)
metric_logging.register_logger(neptune_logger)
except KeyError:
print('HINT: To run with Neptune logging please set your '
'NEPTUNE_API_TOKEN environment variable')
gin.parse_config_files_and_bindings(args.config_file, gin_bindings)
runner = Runner(args.output_dir)
runner.run()