joshvm commited on
Commit
71445f9
·
verified ·
1 Parent(s): 38e6763

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. config.json +1 -59
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,61 +1,3 @@
1
  {
2
- "accelerator": "auto",
3
- "annotations_xml": null,
4
- "architecture": "deformable_detr",
5
- "batch_size": 1,
6
- "devices": "auto",
7
- "model": {
8
- "name": "joshvm/deepforest-detr",
9
- "revision": "main"
10
- },
11
- "nms_thresh": 0.05,
12
- "num_classes": 1,
13
- "patch_overlap": 0.05,
14
- "patch_size": 400,
15
- "path_to_raster": null,
16
- "path_to_rgb": null,
17
- "predict": {
18
- "pin_memory": false
19
- },
20
- "rgb_dir": null,
21
- "score_thresh": 0.25,
22
- "train": {
23
- "csv_file": null,
24
- "epochs": 1,
25
- "fast_dev_run": false,
26
- "lr": 0.001,
27
- "preload_images": false,
28
- "root_dir": null,
29
- "scheduler": {
30
- "params": {
31
- "T_max": 10,
32
- "cooldown": 0,
33
- "eps": 1e-08,
34
- "eta_min": 1e-05,
35
- "factor": 0.1,
36
- "gamma": 0.1,
37
- "lr_lambda": "0.95 ** epoch",
38
- "milestones": [
39
- 50,
40
- 100
41
- ],
42
- "min_lr": 0,
43
- "mode": "min",
44
- "patience": 10,
45
- "step_size": 30,
46
- "threshold": 0.0001,
47
- "threshold_mode": "rel"
48
- },
49
- "type": null
50
- }
51
- },
52
- "validation": {
53
- "csv_file": null,
54
- "iou_threshold": 0.4,
55
- "preload_images": false,
56
- "root_dir": null,
57
- "size": null,
58
- "val_accuracy_interval": 20
59
- },
60
- "workers": 0
61
  }
 
1
  {
2
+ "architecture": "deformable_detr"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:db856097c5c36944c63563e5fc32ff9d8ef4e80c15b4e83650e1523a788bdfab
3
  size 160684940
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2ec08d35171a9c088654b6199ecf42782fe95c778ac500c7dd672dbebf6c08e
3
  size 160684940