Error in write_model

[israrbacha]

Hi,
I try your code but after 300 epoch it give me error in model writer the original message is :
TypeError: write_model() missing 1 positional argument: 'model'

I didn't change your code everything is as it is. I don't know whats the problem the error is in
line 62 in train:
saver.write_model(-1,model)
in epoch 302 when it reaches to 183 iterations it stop with the above message.
please have a look >>>> thanks

[haruka0000]

Hi. I also got same error. I think it should fix like below.
saver.write_model(-1, model)
saver.write_model(-1, total_it, model)
It needs 3 args.