[example] Enable PyTorch for some training example

[frankfliu]

Description

Brief description of what this PR is about

• If this change is a backward incompatible change, why must this change be made?
• Interesting edge cases to note here

[codecov-commenter]

:warning: Please install the to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 58.24%. Comparing base (d7c8a74) to head (b959e2c). Report is 75 commits behind head on master.

:exclamation: Your organization needs to install the Codecov GitHub app to enable full functionality.

:exclamation: There is a different number of reports uploaded between BASE (d7c8a74) and HEAD (b959e2c). Click for more details.

HEAD has 2 uploads less than BASE

| Flag | BASE (d7c8a74) | HEAD (b959e2c) | |------|------|------| ||3|1|

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #3398 +/- ## ============================================= - Coverage 72.86% 58.24% -14.63% + Complexity 7352 5736 -1616 ============================================= Files 668 668 Lines 32509 32509 Branches 3439 3439 ============================================= - Hits 23688 18934 -4754 - Misses 7207 12125 +4918 + Partials 1614 1450 -164 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.