Nathan Lambert
commited on
Commit
·
1d96e57
1
Parent(s):
d5c9c3d
Update config.json
Browse files- config.json +7 -5
config.json
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
-
{
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
|
|
|
|
|
|
|
|
| 1 |
+
{"_class_name": "TemporalUNet",
|
| 2 |
+
"horizon": 32,
|
| 3 |
+
"dim": 32,
|
| 4 |
+
"dim_mults": [1, 4, 8],
|
| 5 |
+
"transition_dim": 14,
|
| 6 |
+
"cond_dim": 3
|
| 7 |
+
}
|