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)
