search

HsinYingLee / DRIT

Learning diverse image-to-image translation from unpaired data
844 stars 155 forks source link

Error in write_model #39

Open israrbacha opened 4 years ago

israrbacha commented 4 years ago

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 commented 4 years ago

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.