We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ab8e47d commit 6876a0fCopy full SHA for 6876a0f
scripts/make_scene_samples.py
@@ -52,7 +52,7 @@ def get_z_indices_crop_out(z_indices: Tensor, predict_x: int, predict_y: int) ->
52
53
@torch.no_grad()
54
def sample(model: Net2NetTransformer, annotations: List[Annotation], dataset: AnnotatedObjectsDataset,
55
- conditional_builder: CoordinatesCenterPointsConditionalBuilder, no_samples: int,
+ conditional_builder: ObjectsCenterPointsConditionalBuilder, no_samples: int,
56
temperature: float, top_k: int) -> Tensor:
57
x_max, y_max = desired_z_shape[1], desired_z_shape[0]
58
0 commit comments