JerryLiJinyi's picture
Upload 127 files
10b912d verified
raw
history blame
965 Bytes
{
"loader": "loaders/newsroom.py",
"dataset": "data/train-data/newsroom",
"indices": "data/train-data/newsroom/indices.npy",
"model_dir": "data/models/newsroom-L11",
"verbose": true,
"print_every": 1,
"eval_every": 50,
"save_every": 50,
"max_val_steps": 512,
"max_train_seconds": null,
"max_train_steps": 8000,
"batch_size": 4,
"learning_rate": 1e-05,
"k_samples": 100,
"sample_aggregation": "max",
"loss": "pgb",
"encoder_model_id": "distilroberta-base",
"rewards": {
"Fluency": {
"weight": 1,
"type": "masked",
"model_id": "distilroberta-base",
"max_score": 40.0,
"norm": "max"
},
"BiEncoderSimilarity": {
"weight": 1,
"model_id": "all-distilroberta-v1"
},
"GaussianLength": {
"weight": 1,
"mean": 11,
"std": 4.4
}
}
}