tbp.monty.frameworks.loggers

tbp.monty.frameworks.loggers.exp_logger

class BaseMontyLogger(handlers)

Bases: object

Basic logger that logs or saves information when logging is called.

close(logger_args, output_dir, model)

flush()

post_episode(logger_args, output_dir, model)

post_epoch(logger_args, output_dir, model)

post_eval(logger_args, output_dir, model)

post_step(logger_args, output_dir, model)

post_train(logger_args, output_dir, model)

pre_episode(logger_args, output_dir, model)

pre_epoch(logger_args, output_dir, model)

pre_eval(logger_args, output_dir, model)

pre_step(logger_args, output_dir, model)

pre_train(logger_args, output_dir, model)

class LoggingCallbackHandler(loggers, model, output_dir)

Bases: object

Calls a list of loggers on an event (eg post_train).

Each logger receives: logger_args: dict with time stamps (steps, epochs, etc.) and

dataloader.primary_target which contains object id and pose

output_dir: Full path of the directory to store log files

Note

This logger handler is intended primarily for logging

call_event(event, logger_args)

close(logger_args)

post_episode(logger_args)

post_epoch(logger_args)

post_eval(logger_args)

post_step(logger_args)

post_train(logger_args)

pre_episode(logger_args)

pre_epoch(logger_args)

pre_eval(logger_args)

pre_step(logger_args)

pre_train(logger_args)

property logger_list

class TestLogger(handlers)

Bases: BaseMontyLogger

close(logger_args, output_dir, model)

post_episode(logger_args, output_dir, model)

post_epoch(logger_args, output_dir, model)

post_eval(logger_args, output_dir, model)

post_train(logger_args, output_dir, model)

pre_episode(logger_args, output_dir, model)

pre_epoch(logger_args, output_dir, model)

pre_eval(logger_args, output_dir, model)

pre_train(logger_args, output_dir, model)

tbp.monty.frameworks.loggers.graph_matching_loggers

class BasicGraphMatchingLogger(handlers)

Bases: BaseMontyLogger

Basic logger that logs or saves information when logging is called.

flush()

get_formatted_overall_stats(mode, episode)

log_episode(logger_args, output_dir, model)

maybe_log(logger_args, output_dir, model)

Left here in case we go back to size based logging.

Remove if not used after code has stabilized.

post_episode(logger_args, output_dir, model)

update_episode_data(logger_args, output_dir, model)

Run get_stats_per_lm and add to overall stats.

Store stats ~

1 (episode)

lm_0 (which lm)

stats

update_overall_stats(mode, episode, episode_steps, monty_matching_steps)

Update overall run stats for mode.

class DetailedGraphMatchingLogger(handlers)

Bases: BasicGraphMatchingLogger

Log detailed stats as .json file by saving data for each LM and SM.

log_episode(logger_args, output_dir, model)

post_episode(logger_args, output_dir, model)

update_episode_data(logger_args, output_dir, model)

Add episode data to overall buffer_data dict.

Store stats ~

1 (episode)

lm_0 (which lm)

stats

class SelectiveEvidenceLogger(handlers)

Bases: BasicGraphMatchingLogger

Log evidences as .json file by saving data for each LM and SM.

This is slimmed down to only log data needed for object similarity analysis. Data logged:

• evidences for each object and pose at the end of an episode

• first frame of the view finder

log_episode(logger_args, output_dir, model)

post_episode(logger_args, output_dir, model)

update_episode_data(logger_args, output_dir, model)

Add episode data to overall buffer_data dict.

tbp.monty.frameworks.loggers.monty_handlers

class BasicCSVStatsHandler

Bases: MontyHandler

Grab any logs at the BASIC level and append to train or eval CSV files.

close()

classmethod log_level()

Handlers filter information from the data they receive.

This class method specifies the level they filter at.

move_columns_to_front(df, columns)

report_episode(data, output_dir, episode, mode='train', **kwargs)

class DetailedJSONHandler

Bases: MontyHandler

Grab any logs at the DETAILED level and append to a json file.

close()

classmethod log_level()

Handlers filter information from the data they receive.

This class method specifies the level they filter at.

report_episode(data, output_dir, episode, mode='train', **kwargs)

Report episode data.

Changed name to report episode since we are currently running with reporting and flushing exactly once per episode.

class MontyHandler

Bases: object

abstract close()

abstract classmethod log_level()

Handlers filter information from the data they receive.

This class method specifies the level they filter at.

abstract report_episode(**kwargs)

class ReproduceEpisodeHandler

Bases: MontyHandler

close()

classmethod log_level()

Handlers filter information from the data they receive.

This class method specifies the level they filter at.

report_episode(data, output_dir, episode, mode='train', **kwargs)

tbp.monty.frameworks.loggers.wandb_handlers