def test_train_localizer(tub, tub_path): t = Tub(tub_path) assert t is not None import donkeycar.templates.cfg_complete as cfg tempfolder = tub_path[:-3] model_path = os.path.join(tempfolder, 'test.h5') cfg_defaults(cfg) tub = tub_path model = model_path transfer = None model_type = "localizer" continuous = False aug = False multi_train(cfg, tub, model, transfer, model_type, continuous, aug)
def test_train_seq(tub, tub_path): t = Tub(tub_path) assert t is not None import donkeycar.templates.cfg_complete as cfg tempfolder = tub_path[:-3] model_path = os.path.join(tempfolder, 'test.h5') cfg.MAX_EPOCHS = 1 cfg.BATCH_SIZE = 10 cfg.SHOW_PLOT = False cfg.VEBOSE_TRAIN = False cfg.OPTIMIZER = "adam" tub = tub_path model = model_path transfer = None model_type = "rnn" continuous = False aug = True multi_train(cfg, tub, model, transfer, model_type, continuous, aug)