[training] set rest of the blocks with requires_grad False. (#10607)

[training] set rest of the blocks with requires_grad False. (#10607) #18818