Hugging Face's logo

crop-classification
/
messis

Transformers
Safetensors
Crop classification
Geospatial foundation models
Prithvi
Fine-tuning
Remote sensing
ZueriCrop 2.0 dataset
Hierarchical classification
Model card Files
xet
Community
yvokeller commited on
Commit
2a97f13
·
verified ·
1 Parent(s): a11655d

Messis | W&B Run exp-2-optimizer-lr-sgd-momentum-4 (https://wandb.ai/crop-classification/messis/runs/vss1692r)

Browse files
Files changed (2) hide show
  1. config.json +60 -55
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,56 +1,61 @@
1
- {
2
- "hparams": {
3
- "accumulate_grad_batches": 8,
4
- "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
5
- "bands": [
6
- 0,
7
- 1,
8
- 2,
9
- 3,
10
- 4,
11
- 5
12
- ],
13
- "batch_size": 2,
14
- "debug": false,
15
- "early_stopping_patience": 10,
16
- "experiment_group": "exp-3-timeranges",
17
- "experiment_name": "exp-3-timeranges-april-august",
18
- "freeze_backbone": true,
19
- "img_size": 224,
20
- "lr": 0.001,
21
- "max_epochs": 100,
22
- "name": "april-august",
23
- "num_frames": 3,
24
- "patch_size": 16,
25
- "test_folds": [
26
- 5
27
- ],
28
- "tiers": {
29
- "tier1": {
30
- "loss_weight": 1,
31
- "num_classes": 6
32
- },
33
- "tier2": {
34
- "loss_weight": 1,
35
- "num_classes": 17
36
- },
37
- "tier3": {
38
- "loss_weight": 1,
39
- "num_classes": 49
40
- },
41
- "tier3_refined": {
42
- "loss_weight": 1,
43
- "num_classes": 49
44
- }
45
- },
46
- "train_folds": [
47
- 0,
48
- 1,
49
- 2,
50
- 4
51
- ],
52
- "val_folds": [
53
- 3
54
- ]
55
- }
 
 
 
 
 
56
  }
 
1
+ {
2
+ "hparams": {
3
+ "accumulate_grad_batches": 4,
4
+ "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
5
+ "bands": [
6
+ 0,
7
+ 1,
8
+ 2,
9
+ 3,
10
+ 4,
11
+ 5
12
+ ],
13
+ "batch_size": 4,
14
+ "debug": false,
15
+ "early_stopping_metric": "val_loss",
16
+ "early_stopping_mode": "min",
17
+ "early_stopping_patience": 10,
18
+ "experiment_group": "exp-2-optimizer-lr",
19
+ "experiment_name": "exp-2-optimizer-lr-sgd-momentum",
20
+ "freeze_backbone": true,
21
+ "img_size": 224,
22
+ "lr": 0.001,
23
+ "max_epochs": 400,
24
+ "name": "sgd-momentum",
25
+ "num_frames": 3,
26
+ "optimizer": "SGD",
27
+ "optimizer_momentum": 0.9,
28
+ "optimizer_weight_decay": null,
29
+ "patch_size": 16,
30
+ "test_folds": [
31
+ 5
32
+ ],
33
+ "tiers": {
34
+ "tier1": {
35
+ "loss_weight": 1,
36
+ "num_classes": 6
37
+ },
38
+ "tier2": {
39
+ "loss_weight": 1,
40
+ "num_classes": 17
41
+ },
42
+ "tier3": {
43
+ "loss_weight": 1,
44
+ "num_classes": 49
45
+ },
46
+ "tier3_refined": {
47
+ "loss_weight": 1,
48
+ "num_classes": 49
49
+ }
50
+ },
51
+ "train_folds": [
52
+ 0,
53
+ 1,
54
+ 2,
55
+ 3
56
+ ],
57
+ "val_folds": [
58
+ 4
59
+ ]
60
+ }
61
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1472a5672c99ceab0291ebbbb5befba3ada45eece8fbaa9266f66853014d9f41
3
  size 713881052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb1e240ff3eb9796d9294fafcd9e7328f9bc71a4adac797b92726097137f600a
3
  size 713881052