Skip to content

Inference Move API#178

Merged
nishadgothoskar merged 1 commit into
gen3dfrom
fiddle_with_inference
Sep 16, 2024
Merged

Inference Move API#178
nishadgothoskar merged 1 commit into
gen3dfrom
fiddle_with_inference

Conversation

@nishadgothoskar

Copy link
Copy Markdown
Collaborator

No description provided.

@nishadgothoskar
nishadgothoskar merged commit 4b8e6fe into gen3d Sep 16, 2024
@nishadgothoskar
nishadgothoskar deleted the fiddle_with_inference branch September 16, 2024 17:59
key = b3d.split_key(key)
trace = inference_step(trace, key, all_data[T]["rgbd"])
trace = inference.advance_time(key, trace, all_data[T]["rgbd"])
trace = inference_old.inference_step(trace, key, inference_hyperparams)[

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@nishadgothoskar for tonight, can we aim to launch at least 1 run that is using the new inference algorithm? This will be through a call to inference.inference_step_c2f. I can add this to the script if helpful.

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yes definitely. that is the plan.

key,
observed_rgbd_for_point=observed_rgbd_for_point,
latent_depth=new_state["pose"].apply(hyperparams["vertices"][vertex_index])[2],
latent_rgbd_for_point=jnp.array(

@georgematheos georgematheos Sep 16, 2024

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@nishadgothoskar isn't this going to break inference?

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What I mean is -- don't we need to set the rgb correctly here?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i think the rgb is ignored

@georgematheos

georgematheos commented Sep 16, 2024 via email

Copy link
Copy Markdown
Collaborator

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants