Set args.admm as True and args.masked_retrain as False in main.py to run admm function.
Set args.admm as False and args.masked_retrain as True in main.py to run masked_retrain function.
files_main.py can continuously run several YAML files as long as name those files following a specific rule and put those into seg_static_configs directory