The purpose of gen_test object in FUNITModel

[NTU-P04922004]

The gen_test object in FUNITModel seems to be a clone of the whole model except the discriminator. The gen_test object is only used in generating testing results and it seems that it is redundant because we can run the gen object in evaluation mode. Why making a clone of the gen object? Was it designed for the multi-GPU environment?

FUNIT/funit_model.py

Line 23 in 4cd5f22

 self.gen_test = copy.deepcopy(self.gen) 

[GU Zheng]

@NTU-P04922004

I find the answer in the paper.

The final generator is a historical average version of the intermediate generators where the update weight is 0.001.

FUNIT/trainer.py

Line 18 in 4cd5f22

 def update_average(model_tgt, model_src, beta=0.999): 

[NTU-P04922004]

@NTU-P04922004

I find the answer in the paper.

The final generator is a historical average version of the intermediate generators where the update weight is 0.001.

FUNIT/trainer.py

Line 18 in 4cd5f22

 def update_average(model_tgt, model_src, beta=0.999): 

Got it, thanks!