Fix transformer and error on example when CI uses single-GPU (#757)

Fix transformer and error on example when CI uses single-GPU (#757) #405