
my code:
plt.hist(all_loss[0][-1][clean_idx].numpy(), bins=20,density=True,alpha=0.5,stacked=True,facecolor="cornflowerblue",label='clean')
plt.hist(all_loss[0][-1][noisy_idx].numpy(), bins=20, density=True,alpha=0.5,stacked=True,facecolor="pink",label='noisy')
The picture I get:

