src/fine_tune.py

import draccus
import torch
import yaml
from dataclasses import asdict
from iba.configIBA import MainConfig

@draccus.wrap(config_path="./config_yaml/config.yaml")
def main(mainCfg: MainConfig):
    print('Hello\n', mainCfg)
    cfg_dict = asdict(mainCfg)
    print(yaml.dump(cfg_dict, indent=2, default_flow_style=False))

    

if __name__ == "__main__":
    main()