(with_tensorboard, with_visualizer, experiment_name, viewer_config_path=None)
| 6 | |
| 7 | |
| 8 | def create_callbacks(with_tensorboard, with_visualizer, experiment_name, viewer_config_path=None): |
| 9 | cb_list = [] |
| 10 | if(with_tensorboard): |
| 11 | from callbacks.tensorboard_callback import TensorboardCallback #we put it here in case we don't have tensorboard installed |
| 12 | tensorboard_callback=TensorboardCallback(experiment_name) |
| 13 | cb_list.append(tensorboard_callback) |
| 14 | if(with_visualizer): |
| 15 | from callbacks.viewer_callback import ViewerCallback #we put it here because we might not have the visualizer package installed |
| 16 | viewer_callback=ViewerCallback(viewer_config_path, experiment_name) |
| 17 | cb_list.append(viewer_callback) |
| 18 | cb_list.append(StateCallback()) |
| 19 | cb = CallbacksGroup(cb_list) |
| 20 | |
| 21 | return cb |
no test coverage detected