default_config_files/learning/active_learning_LSTM.json

{
    "train_dataloader": {
        "batch_size": 512,
        "shuffle": true,
        "pin_memory": true,
        "drop_last": true,
        "num_workers": 0
    },
    "val_dataloader": {},
    "trainer": {
        "accelerator": "gpu",
        "devices": 1,
        "max_epochs": 50,
        "check_val_every_n_epoch": 51
    },
    "trainer_fit_kwargs": {}
}