main_local_debug.py

from main import ex
import yaml
import torch


torch.autograd.set_detect_anomaly(True, check_nan=True)



with open("configs_local_debug/GCN/upfd_pol_transfer.yaml") as f:
    cfg = yaml.safe_load(f)

r = ex.run(config_updates=cfg)