Spaces:
Runtime error
Runtime error
File size: 764 Bytes
10b912d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
{
"loader": "loaders/gigaword.py",
"dataset": "data/train-data/gigaword",
"indices": "data/train-data/gigaword/indices.npy",
"model_dir": "data/models/example",
"verbose": true,
"print_every": 1,
"eval_every": 10,
"save_every": 10,
"max_val_steps": 8,
"max_train_seconds": null,
"max_train_steps": 1000,
"batch_size": 1,
"learning_rate": 1e-05,
"k_samples": 10,
"sample_aggregation": "max",
"loss": "pgb",
"encoder_model_id": "distilroberta-base",
"rewards": {
"BiEncoderSimilarity": {
"weight": 1,
"model_id": "all-distilroberta-v1"
},
"GaussianCR": {
"weight": 1,
"mean": 0.5,
"std": 0.2
}
}
}
|