Skip to content

Commit

Permalink
fix in data loader debug function
Browse files Browse the repository at this point in the history
  • Loading branch information
Gregor Ramien committed Apr 18, 2020
1 parent 687ba14 commit 3b87628
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions experiments/toy_exp/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,13 +290,14 @@ def copy_and_unpack_data(logger, pids, fold_dir, source_dir, target_dir):

if __name__=="__main__":
import utils.exp_utils as utils
from .configs import Configs
cf_file = utils.import_module("cf", "configs.py")

total_stime = time.time()


cf = Configs()
logger = utils.get_logger(0)
cf = cf_file.configs(server_env=False)
cf.server_env = False
logger = utils.get_logger(".")
batch_gen = get_train_generators(cf, logger)

train_batch = next(batch_gen["train"])
Expand Down

0 comments on commit 3b87628

Please sign in to comment.