# Text-heavy configuration — stress test of caption contribution.
# L = 0.5 * L_img_img + 0.5 * L_sat_cap + 0.5 * L_drone_cap + 0.2 * L_cap_cap

include 'balanced.gin'

MultiTermInfoNCE.lambda_ii = 0.5
MultiTermInfoNCE.lambda_sc_max = 0.5
MultiTermInfoNCE.lambda_dc_max = 0.5
MultiTermInfoNCE.lambda_cc_max = 0.2

TrainConfig.output_dir = "out/caption_test/text_heavy"