| import plotly.express as px | |
| def plot_curvature(curv): | |
| layers=list(curv.keys()) | |
| vals=list(curv.values()) | |
| fig=px.bar(x=layers,y=vals,title="Curvature per Layer") | |
| return fig | |
| def plot_precision_plan(plan): | |
| layers=list(plan.keys()) | |
| vals=list(plan.values()) | |
| fig=px.bar(x=layers,y=[1 if v=='fp16' else 0 for v in vals], | |
| title="Precision Plan (1=FP16, 0=FP32)") | |
| return fig | |