File size: 415 Bytes
e987e3f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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
|