praeclarumjj3 commited on
Commit
468c569
1 Parent(s): fad7dff
Files changed (2) hide show
  1. config.json +14 -1
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -24,7 +24,7 @@
24
  ],
25
  "diversity_penalty": 0.0,
26
  "do_sample": false,
27
- "drop_path_rate": 0.1,
28
  "early_stopping": false,
29
  "embed_dim": 192,
30
  "encoder_no_repeat_ngram_size": 0,
@@ -2062,6 +2062,12 @@
2062
  ],
2063
  "num_layers": 4,
2064
  "num_return_sequences": 1,
 
 
 
 
 
 
2065
  "output_attentions": false,
2066
  "output_hidden_states": false,
2067
  "output_scores": false,
@@ -2077,6 +2083,13 @@
2077
  "return_dict": true,
2078
  "return_dict_in_generate": false,
2079
  "sep_token_id": null,
 
 
 
 
 
 
 
2080
  "suppress_tokens": null,
2081
  "task_specific_params": null,
2082
  "temperature": 1.0,
 
24
  ],
25
  "diversity_penalty": 0.0,
26
  "do_sample": false,
27
+ "drop_path_rate": 0.3,
28
  "early_stopping": false,
29
  "embed_dim": 192,
30
  "encoder_no_repeat_ngram_size": 0,
 
2062
  ],
2063
  "num_layers": 4,
2064
  "num_return_sequences": 1,
2065
+ "out_features": [
2066
+ "stage1",
2067
+ "stage2",
2068
+ "stage3",
2069
+ "stage4"
2070
+ ],
2071
  "output_attentions": false,
2072
  "output_hidden_states": false,
2073
  "output_scores": false,
 
2083
  "return_dict": true,
2084
  "return_dict_in_generate": false,
2085
  "sep_token_id": null,
2086
+ "stage_names": [
2087
+ "stem",
2088
+ "stage1",
2089
+ "stage2",
2090
+ "stage3",
2091
+ "stage4"
2092
+ ],
2093
  "suppress_tokens": null,
2094
  "task_specific_params": null,
2095
  "temperature": 1.0,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5e146d6c4fd780af86ad0c8e3fcdf88a62c696c060ce80439f7b4fb21218c0fd
3
- size 879533065
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e02211a290c7df91a89dc1065db701d83d85f170aaf1de793894d3fa7120371a
3
+ size 879517453