kd.inspect.get_batch_stats |
Return pd.DataFrame containing the batch stats. |
kd.inspect.get_colab_model_overview |
Return pd.DataFrame for displaying the model params, inputs,… |
kd.inspect.get_connection_graph |
Build the graphviz. |
kd.inspect.json_spec_like |
Convert etree.spec_like output to json and displays it in colab form. |
kd.inspect.lower_trainstep |
Returns lowered trainerstep.step. |
kd.inspect.plot_batch |
Display batch images. |
kd.inspect.plot_context |
Display the context structure. |
kd.inspect.plot_schedules |
Overview plot for (nested) dict of schedules. |
kd.inspect.plot_sharding |
Plot sharding. |
kd.inspect.show_trainer_info |
Display various plot on the trainer. |