import os
import sys
import json

from omegaconf import OmegaConf
from hydra.core.global_hydra import GlobalHydra
from hydra import initialize, initialize_config_module, initialize_config_dir, compose

os.environ['ROOT'] = "/home/quang/workspace/traffic_var"
os.environ['DATA_ROOT'] = "/home/quang/datasets/traffic_var"

# initialize hydra config
GlobalHydra.instance().clear()
initialize(config_path="./")
with_decoder_config = compose(config_name='with_decoder.yaml', overrides=["clip_model=ViT-B/16", "rationale_type=0", "val_rationale_type=0"])
original_config = compose(config_name='experiment.yaml', overrides=["clip_model=ViT-B/16", "rationale_type=0", "val_rationale_type=0"])