wukevin commited on
Commit
6cb6d29
·
1 Parent(s): df23159

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