Dynamic Shape

The Rec SDK training framework supports dynamic shape. You can enable this function by referring to the following content.

Before enabling dynamic shape, install the Kernels operator package. For details, see "Installing Kernels" in "Installing CANN" in CANN Software Installation Guide.

Usage Process

To use the dynamic shape function in the Rec SDK training framework, set use_dynamic of the init API to True during framework initialization.

Sample Code

init(use_mpi, rank_id=rank_id, rank_size=rank_size, train_steps=train_steps, eval_steps=eval_steps, prefetch_batch_number=1, use_dynamic=True)

Parent topic: Function Training Process