habitat.profiling.autograd

Module Contents

Classes

AutogradEngine

Emulates the backward pass for a given model output, for timing purposes.

class habitat.profiling.autograd.AutogradEngine(grad_fn_ordering, input_map, initial_inputs)

Emulates the backward pass for a given model output, for timing purposes.

classmethod new_from(operation_output, exclude_accumulate_grad=True)

run_backward()