[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]

⚠️ Please install the to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

All modified and coverable lines are covered by tests ✅

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

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

❗ 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

@@              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     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.