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 Name Job ID Coverage
  Build
passed build_linux_27 #93455
conda-linux

00:03:28

passed build_linux_35 #93456
conda-linux

00:03:35

passed build_linux_36 #93457
conda-linux

00:03:40

 
  Test
passed test_linux_27 #93458
conda-linux

00:07:45

73.0%
passed test_linux_35 #93459
conda-linux

00:09:21

73.0%
passed test_linux_36 #93460
conda-linux

00:09:30

73.0%