# Text-heavy: gate initialized low (more text weight). # query = sigma(0.3) * drone + 0.7 * text include 'balanced.gin' DualEncoderCaptionTest.init_gate = 0.3 GatedFusion.init_gate = 0.3 TrainConfig.output_dir = "out/caption_test/text_heavy"