Upload folder using huggingface_hub
Browse files- config.json +37 -0
- model.safetensors +3 -0
- onnx/model.onnx +3 -0
    	
        config.json
    ADDED
    
    | @@ -0,0 +1,37 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_name_or_path": "edbeeching/decision-transformer-gym-walker2d-medium",
         | 
| 3 | 
            +
              "act_dim": 6,
         | 
| 4 | 
            +
              "action_tanh": true,
         | 
| 5 | 
            +
              "activation_function": "relu",
         | 
| 6 | 
            +
              "architectures": [
         | 
| 7 | 
            +
                "DecisionTransformerModel"
         | 
| 8 | 
            +
              ],
         | 
| 9 | 
            +
              "attn_pdrop": 0.1,
         | 
| 10 | 
            +
              "bos_token_id": 50256,
         | 
| 11 | 
            +
              "embd_pdrop": 0.1,
         | 
| 12 | 
            +
              "eos_token_id": 50256,
         | 
| 13 | 
            +
              "hidden_size": 128,
         | 
| 14 | 
            +
              "initializer_range": 0.02,
         | 
| 15 | 
            +
              "layer_norm_epsilon": 1e-05,
         | 
| 16 | 
            +
              "max_ep_len": 1000,
         | 
| 17 | 
            +
              "model_type": "decision_transformer",
         | 
| 18 | 
            +
              "n_embd": 768,
         | 
| 19 | 
            +
              "n_head": 1,
         | 
| 20 | 
            +
              "n_inner": null,
         | 
| 21 | 
            +
              "n_layer": 3,
         | 
| 22 | 
            +
              "n_positions": 1024,
         | 
| 23 | 
            +
              "reorder_and_upcast_attn": false,
         | 
| 24 | 
            +
              "resid_pdrop": 0.1,
         | 
| 25 | 
            +
              "scale_attn_by_inverse_layer_idx": false,
         | 
| 26 | 
            +
              "scale_attn_weights": true,
         | 
| 27 | 
            +
              "state_dim": 17,
         | 
| 28 | 
            +
              "summary_activation": null,
         | 
| 29 | 
            +
              "summary_first_dropout": 0.1,
         | 
| 30 | 
            +
              "summary_proj_to_labels": true,
         | 
| 31 | 
            +
              "summary_type": "cls_index",
         | 
| 32 | 
            +
              "summary_use_proj": true,
         | 
| 33 | 
            +
              "torch_dtype": "float32",
         | 
| 34 | 
            +
              "transformers_version": "4.41.2",
         | 
| 35 | 
            +
              "use_cache": true,
         | 
| 36 | 
            +
              "vocab_size": 1
         | 
| 37 | 
            +
            }
         | 
    	
        model.safetensors
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:2cd28aa52bb52aeb8dac8e9df3872a5c0974356c00a34ea13b55d216acb431d2
         | 
| 3 | 
            +
            size 3449272
         | 
    	
        onnx/model.onnx
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:e987f0992a2378899facba3d0ee2496c319ade2dd689dffe6814d88bfa7ba850
         | 
| 3 | 
            +
            size 4607511
         | 

