xnn.gnn.models.blocks.set_recompute_radial

xnn.gnn.models.blocks.set_recompute_radial(model, recompute='auto')

Choose when the radial networks of model recompute their hidden layers.

Parameters:

• model (torch.nn.Module) – Any model; every RadialNet in it is set.

• recompute (bool or str) – "auto" (default), True (whenever a gradient is needed) or False (never); see RadialNet.

Returns:

model.

Return type:

torch.nn.Module