Patched the train.py script to accept tensors as input

6 jobs for updates in 13 minutes and 11 seconds (queued for 2 seconds)
Status Job ID Name Coverage
  Build
passed #93455
conda-linux
build_linux_27

00:03:28

passed #93456
conda-linux
build_linux_35

00:03:35

passed #93457
conda-linux
build_linux_36

00:03:40

 
  Test
passed #93458
conda-linux
test_linux_27

00:07:45

73.0%
passed #93459
conda-linux
test_linux_35

00:09:21

73.0%
passed #93460
conda-linux
test_linux_36

00:09:30

73.0%