PyTorch for Jetson

[dusty_nv]

@father please keep discussion in the other topic that you have opened, thank you.

Python3 train_ssd.py --data=data/fruit --model-dir=models/fruit --batch-size=4 --epochs=30

/home/ljx/.local/lib/python3.6/site-packages/torch/nn/_reduction.py:44: UserWarning: size_average and reduce args will be deprecated, please use reduction=‘sum’ instead. warnings.warn(warning.format(ret)) THCudaCheck FAIL file=/local/jetbot/pytorch/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu line=202 error=701 : too many resources requested for launch Traceback (most recent call last): File “train_ssd.py”, line 342, in device=DEVICE, debug_steps=args.debug_steps, epoch=epoch) F…