Sir I did that. Using this code out of main unction.
plt.plot(Loss_train, 'g', label='Training loss')
plt.plot(Loss_test, 'b', label='Testing loss')
plt.title('Training  & Testing loss')
plt.xlabel('Epochs')
plt.ylabel('Loss')
plt.legend()
plt.show()

plt.plot(Acc_train, 'g', label='Training  acc')
plt.plot(Acc_test, 'b', label='Testing  acc')
plt.title('Training  & Testing acc')
plt.xlabel('Epochs')
plt.ylabel('Acc')
plt.legend()
plt.show()

I have saved the tensors Acc_train, Acc_test, Loss_train and Loss_test. The only confusion is, here the epochs are plotted automatically on x -axis?  As I am not giving epoch here. Result for 5 epochs is plotted as below