Add CATH pretrained model
Browse files
cath_pretrained/config.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"attention_probs_dropout_prob": 0.1,
|
| 3 |
+
"classifier_dropout": null,
|
| 4 |
+
"hidden_act": "gelu",
|
| 5 |
+
"hidden_dropout_prob": 0.1,
|
| 6 |
+
"hidden_size": 384,
|
| 7 |
+
"initializer_range": 0.02,
|
| 8 |
+
"intermediate_size": 768,
|
| 9 |
+
"layer_norm_eps": 1e-12,
|
| 10 |
+
"max_position_embeddings": 128,
|
| 11 |
+
"model_type": "bert",
|
| 12 |
+
"num_attention_heads": 12,
|
| 13 |
+
"num_hidden_layers": 12,
|
| 14 |
+
"pad_token_id": 0,
|
| 15 |
+
"position_embedding_type": "relative_key",
|
| 16 |
+
"transformers_version": "4.11.3",
|
| 17 |
+
"type_vocab_size": 2,
|
| 18 |
+
"use_cache": false,
|
| 19 |
+
"vocab_size": 30522
|
| 20 |
+
}
|
cath_pretrained/models/best_by_valid/epoch=1488-step=565820.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:13093a4c5951213ddd222d255bc4c6e6abdc076e65bd23278da9dbce4e787c67
|
| 3 |
+
size 57921427
|
cath_pretrained/training_args.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"angles_definitions": "canonical-full-angles", "max_seq_len": 128, "min_seq_len": 40, "trim_strategy": "randomcrop", "zero_center": true, "timesteps": 1000, "variance_schedule": "cosine", "variance_scale": 1.0, "time_encoding": "gaussian_fourier", "num_hidden_layers": 12, "hidden_size": 384, "intermediate_size": 768, "num_heads": 12, "position_embedding_type": "relative_key", "dropout_p": 0.1, "decoder": "mlp", "gradient_clip": 1.0, "batch_size": 64, "lr": 5e-05, "loss": "smooth_l1", "l2_norm": 0.0, "l1_norm": 0.0, "circle_reg": 0.0, "min_epochs": 10000, "max_epochs": 10000, "early_stop_patience": 0, "lr_scheduler": "LinearWarmup", "use_swa": false, "subset": null, "exhaustive_validation_t": false, "syn_noiser": "", "single_angle_debug": -1, "single_timestep_debug": false, "cpu_only": false, "ngpu": -1, "write_valid_preds": false, "dryrun": true, "multithread": true}
|
cath_pretrained/training_mean_offset.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:40693db4fcbade60d7133e95b816a862ca58f88ab41b3f5a33ea2b1faed6d257
|
| 3 |
+
size 152
|