config.json

{
    "name": "SkipGram on solution tokens",
    "n_gpu": 1,

    "arch": {
        "type": "SkipGramModel",
        "args": {
            "vocab_size": 10000,
            "embed_size": 128
        }
    },
    "data_loader": {
        "type": "TokenDataProducer",
        "args":{
            "n_tokens": 10000,
            "batch_size": 1024,
            "num_neg_samples": 5
        }
    },
    "optimizer": {
        "type": "Adam",
        "args":{
            "lr": 0.003,
            "amsgrad": true
        }
    },
    "trainer": {
        "epochs": 1000,

        "save_dir": "saved/",
        "save_period": 1,
        "verbosity": 2,
        
        "monitor": "min loss",
        "early_stop": 100,

        "tensorboard": true
    },
    "tensorboard" :{
        "enabled": true,
        "modules": ["tensorboardX", "torch.utils.tensorboard"]
    }
}