Set model weights and bias