[patch] get_transforms_cls after update_config_with_checkpoint

[francoishernandez]

The transforms classes were initialized prior to the config update from checkpoint, preventing transparent loading of transforms from model config when finetuning.