reconsruct

This commit is contained in:
tlk-dsg 2021-05-11 16:36:58 +08:00
parent 7ceb1252e7
commit f8acfc1225
2 changed files with 5 additions and 2 deletions

View File

@ -2,10 +2,10 @@ hydra:
run:
# Output directory for normal runs
dir: ./logs/${now:%Y-%m-%d_%H-%M-%S}
dir: ../logs/${now:%Y-%m-%d_%H-%M-%S}
sweep:
# Output directory for sweep runs
dir: ./logs/${now:%Y-%m-%d_%H-%M-%S}
dir: ../logs/${now:%Y-%m-%d_%H-%M-%S}
# Output sub directory for sweep runs.
subdir: ${hydra.job.num}_${hydra.job.id}

View File

@ -1,5 +1,8 @@
import torch
from torch.utils.data import Dataset
import os
import sys
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
from utils import load_pkl