steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New Dataset Generator, How to differentiate the loss function?

Browse Source
This commit is contained in:
Steffen Illium
2020-02-18 21:58:31 +01:00
parent 61c5cb44a0
commit 8424251ca0
13 changed files with 250 additions and 39 deletions

7
main.py

@ -10,7 +10,7 @@ import warnings
from pytorch_lightning import Trainer
from torch.utils.data import DataLoader
from dataset.dataset import TrajData
from dataset.dataset import TrajPairData
from lib.utils.config import Config
from lib.utils.logging import Logger
@ -32,7 +32,8 @@ main_arg_parser.add_argument("--main_seed", type=int, default=69, help="")
# Data Parameters
main_arg_parser.add_argument("--data_worker", type=int, default=10, help="")
main_arg_parser.add_argument("--data_batchsize", type=int, default=100, help="")
main_arg_parser.add_argument("--data_root", type=str, default='../data/rpoot', help="")
main_arg_parser.add_argument("--data_root", type=str, default='/data/', help="")
main_arg_parser.add_argument("--map_root", type=str, default='/res/maps', help="")
# Transformations
main_arg_parser.add_argument("--transformations_to_tensor", type=strtobool, default=False, help="")
@ -65,7 +66,7 @@ config = Config.read_namespace(args)
# TESTING ONLY #
# =============================================================================
hparams = config.model_paramters
dataset = TrajData('data', mapname='tate', alternatives=10000, trajectories=2500)
dataset = TrajPairData('data', mapname='tate', alternatives=10000, trajectories=2500)
dataloader = DataLoader(dataset=dataset.train_dataset, shuffle=True,
batch_size=hparams.data_param.batchsize,
num_workers=hparams.data_param.worker)