houkensjtu
houkensjtu t1_j1bl0my wrote
Reply to comment by AllowFreeSpeech in [P] A self-driving car using Nvidia Jetson Nano, with movement controlled by a pre-trained convolution neural network (CNN) written in Taichi by TaichiOfficial
Everything related to CNN including the backprop was done manually. And we do have plan to update the documentation - sorry for the non-english code comments!
houkensjtu t1_j1bjta9 wrote
Reply to comment by Just_CurioussSss in [P] A self-driving car using Nvidia Jetson Nano, with movement controlled by a pre-trained convolution neural network (CNN) written in Taichi by TaichiOfficial
You're right that Taichi is for general numerical computations. We just wanted to explore the possibility of making a CNN in Taichi - it's easier to do with Pytorch for sure, but I guess that kind of "uselessness" is part of the purpose of a "Hackathon" www
Also, check out this blog for more detailed comparison: https://docs.taichi-lang.org/blog/taichi_vs_torch_01
houkensjtu t1_j1gcwls wrote
Reply to comment by Dry_Task4749 in [P] A self-driving car using Nvidia Jetson Nano, with movement controlled by a pre-trained convolution neural network (CNN) written in Taichi by TaichiOfficial
I've done some comparisons before, and the conclusion is that it really depends on the computation task. Generally speaking, Numba on CPU is very similar to Taichi in terms of concepts and performance. Numba on GPU is more like writing CUDA, while Taichi allows you to use the same code for CPU & GPU.
You can refer to this repo for more detailed benchmark results: https://github.com/taichi-dev/taichi_benchmark/tree/main/suites