shunk031 commited on
Commit
793deb3
1 Parent(s): c34256f

deploy: 7d699a4c61838f1596e1cfa9c53f636c1c4126e5

Browse files
Files changed (1) hide show
  1. layout-alignment.py +2 -2
layout-alignment.py CHANGED
@@ -20,7 +20,7 @@ Returns:
20
  Examples:
21
 
22
  Example 1: Single processing
23
- >>> metric = evaluate.load("pytorch-layout-generation/layout-alignment")
24
  >>> model_max_length, num_coordinates = 25, 4
25
  >>> bbox = np.random.rand(model_max_length, num_coordinates)
26
  >>> mask = np.random.choice(a=[True, False], size=(model_max_length,))
@@ -28,7 +28,7 @@ Examples:
28
  >>> print(metric.compute())
29
 
30
  Example 2: Batch processing
31
- >>> metric = evaluate.load("pytorch-layout-generation/layout-alignment")
32
  >>> batch_size, model_max_length, num_coordinates = 512, 25, 4
33
  >>> batch_bbox = np.random.rand(batch_size, model_max_length, num_coordinates)
34
  >>> batch_mask = np.random.choice(a=[True, False], size=(batch_size, model_max_length))
 
20
  Examples:
21
 
22
  Example 1: Single processing
23
+ >>> metric = evaluate.load("creative-graphic-design/layout-alignment")
24
  >>> model_max_length, num_coordinates = 25, 4
25
  >>> bbox = np.random.rand(model_max_length, num_coordinates)
26
  >>> mask = np.random.choice(a=[True, False], size=(model_max_length,))
 
28
  >>> print(metric.compute())
29
 
30
  Example 2: Batch processing
31
+ >>> metric = evaluate.load("creative-graphic-design/layout-alignment")
32
  >>> batch_size, model_max_length, num_coordinates = 512, 25, 4
33
  >>> batch_bbox = np.random.rand(batch_size, model_max_length, num_coordinates)
34
  >>> batch_mask = np.random.choice(a=[True, False], size=(batch_size, model_max_length))