DRIT
DRIT copied to clipboard
Error in write_model
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
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.