Невозможно удалить предыдущий сюжет в matplotlib

Question

Я хочу построить график предсказанной функции в сравнении с исходной функцией в matplotlib.Я использовал следующий код:

import matplotlib as plt

plt.ion()
for i in range(200):
    #Some Operations

    if i%5==0 :
        print(str((i+1))+'th loss',loss.squeeze())
        # plot and show learning process
        plt.cla()
        plt.scatter(x.data.numpy(), y.data.numpy())
        plt.plot(x.data.numpy(), output.data.numpy(), 'r-', lw=5)
        plt.text(0.5, 0, 'Loss=%.4f' % loss.data[0], fontdict={'size': 20, 'color':  'red'})
        plt.pause(0.1)

plt.ioff()
plt.show()

При выполнении этого кода результирующий график обновляется, но отображается поверх предыдущего графика, хотя я использовал функцию cla() для очистки.
Весь кодисполняемый код можно найти здесь .