Create asgard.yaml
Browse files- asgard.yaml +12 -0
asgard.yaml
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
weights:
|
| 2 |
+
_target_: asgard.models.api.Weights
|
| 3 |
+
url: s3://h3-artifacts/test/tiny_test_model_state_dict.pth
|
| 4 |
+
region_name: eu-central-2
|
| 5 |
+
endpoint_url: https://s3.eu-central-2.wasabisys.com
|
| 6 |
+
meta:
|
| 7 |
+
model_config:
|
| 8 |
+
_target_: asgard.models._test.TinyTestModel
|
| 9 |
+
input_dim: 4
|
| 10 |
+
hidden_dim: 8
|
| 11 |
+
output_dim: 2
|
| 12 |
+
|