github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-26 21:10:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix visualization

Browse Source
This commit is contained in:
YeAnbang 2025-06-20 15:52:09 +08:00
parent c2561f826a
commit db8baeeaf2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
applications/ColossalChat/visualization.py
View File

@ -74,8 +74,8 @@ for idx, (actor, func_dict) in enumerate(actors.items()):
yticks.append(y_val) yticks.append(y_val)
yticklabels.append(f"{actor}:{func}") yticklabels.append(f"{actor}:{func}")
for start, end in intervals: for start, end in intervals:
if end - start < 6: if end - start < 1:
end = start + 6 # Ensure minimum length of 100ms end = start + 1 # Ensure all lines are at least 3 units long
ax.plot( ax.plot(
[start, end], [start, end],
[y_val, y_val], [y_val, y_val],
@ -96,5 +96,5 @@ unique = dict(zip(labels, handles))
ax.legend(unique.values(), unique.keys()) ax.legend(unique.values(), unique.keys())
plt.tight_layout() plt.tight_layout()
plt.grid(True) plt.grid(True)
plt.savefig(args.visualization) plt.savefig(args.visualization, dpi=600) # Increase dpi for higher resolution
print(f"Plot saved as {args.visualization}") print(f"Plot saved as {args.visualization}")