Trying out TensorRT on Jetson TX2

As I was learning how to use TensorRT to achieve best inference performance on Jetson TX2, I used my previously trained cats-dogs Caffe model for experiment. The result was good, as expected.

Share "Trying out TensorRT on Jetson TX2"

Share on: FacebookTwitter

Deep Learning Cats Dogs Tutorial on Jetson TX2

I ran the Deep Leanring Cats Dogs Tutorial code to train an AlexNet on Jetson TX2. This is the best Caffe and Pyhton tutorial I've come across so far. I highly recommend it to people who wants to learn Caffe.

Share "Deep Learning Cats Dogs Tutorial on Jetson TX2"

Share on: FacebookTwitter

Nintendo AI Agent Training in Action, Finally...

I finally managed to put everything together, and started training my AI Agent (DeepMind's DQN) to play 'Galaga' on Nintendo Famicom Mini. Hopefully the AI could learn in a couple of weeks to play significantly better than the random player so that I could report the progress then...

Share "Nintendo AI Agent Training in Action, Finally..."

Share on: FacebookTwitter

Letting a Random Agent Play Galaga

A random agent (player) playing Galaga can get an average score of 6314.3! Hopefully my AI agent could do much better than that after training.

Share "Letting a Random Agent Play Galaga"

Share on: FacebookTwitter

Galaga Game Environment

I integrated the vidcap, galaga and gpio modules and finalized a Galaga game environment API which is similar to xtari, the Atari game emulator.

Share "Galaga Game Environment"

Share on: FacebookTwitter

Using cuDNN to Speed Up DQN Training on Jetson TX1

I found it was quite easy to enable cudnn in Torch7. I was able to reduce DQN training time by 1/3 on Jetson TX1 with change of only a few lines of code.

Share "Using cuDNN to Speed Up DQN Training on Jetson TX1"

Share on: FacebookTwitter

blog built using the cayman-theme by Jason Long. LICENSE