This commit is contained in:
游雁 2023-12-06 19:55:24 +08:00
parent 81acb17544
commit d6aa93946e

View File

@ -37,7 +37,7 @@ def preprocess_config(cfg: DictConfig):
@hydra.main()
def main(kwargs: DictConfig):
# preprocess_config(kwargs)
import pdb; pdb.set_trace()
# import pdb; pdb.set_trace()
# set random seed
set_all_random_seed(kwargs.get("seed", 0))
torch.backends.cudnn.enabled = kwargs.get("cudnn_enabled", torch.backends.cudnn.enabled)