JaMe76 commited on
Commit
8bc5c6f
1 Parent(s): 1f0867b

update how it was trained

Browse files
Files changed (1) hide show
  1. README.md +14 -1
README.md CHANGED
@@ -43,8 +43,21 @@ dataset_val = publaynet
43
  build_val_config = ["max_datapoints=2000"]
44
 
45
  coco_metric = MetricRegistry.get_metric("coco")
46
- ```
47
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  ## How to fine-tune this model
49
 
50
  To fine tune this model, please check this [Fine-tune](https://github.com/deepdoctection/deepdoctection/blob/master/notebooks/Fine_Tune.ipynb) tutorial.
 
43
  build_val_config = ["max_datapoints=2000"]
44
 
45
  coco_metric = MetricRegistry.get_metric("coco")
 
46
 
47
+
48
+ train_faster_rcnn(path_config_yaml=path_config_yaml,
49
+ dataset_train=dataset_train,
50
+ path_weights=path_weights,
51
+ config_overwrite=config_overwrite,
52
+ log_dir="/path/to/dir",
53
+ build_train_config=build_train_config,
54
+ dataset_val=dataset_val,
55
+ build_val_config=build_val_config,
56
+ metric=coco_metric,
57
+ pipeline_component_name="ImageLayoutService"
58
+ )
59
+ ```
60
+
61
  ## How to fine-tune this model
62
 
63
  To fine tune this model, please check this [Fine-tune](https://github.com/deepdoctection/deepdoctection/blob/master/notebooks/Fine_Tune.ipynb) tutorial.