Upload config.yaml with huggingface_hub
Browse files- config.yaml +6 -1
config.yaml
CHANGED
|
@@ -53,7 +53,12 @@ learner:
|
|
| 53 |
train_iters: 10000
|
| 54 |
trainers: null
|
| 55 |
val_dataloaders: null
|
| 56 |
-
model:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
num_workers: 4
|
| 58 |
optimizer:
|
| 59 |
_partial_: true
|
|
|
|
| 53 |
train_iters: 10000
|
| 54 |
trainers: null
|
| 55 |
val_dataloaders: null
|
| 56 |
+
model:
|
| 57 |
+
_target_: mlproject.models.config.build_tsm_model
|
| 58 |
+
model_name: tsm_resnet50
|
| 59 |
+
num_classes: 51
|
| 60 |
+
num_frames: 8
|
| 61 |
+
pretrained: kinetics400
|
| 62 |
num_workers: 4
|
| 63 |
optimizer:
|
| 64 |
_partial_: true
|