working on -> https://github.com/naruya/tecnets-pytorch/tree/mini/loadall
(↑Fast mode. very powerful GPU is needed.)
Original Paper: Task-Embedded Control Networks for Few-Shot Imitation Learning (Stephen James, Michael Bloesch, Andrew J. Davison)
sim_push success rate: 33.3%...
batch_task 64, n_support_demo 6.
(because of my tiny bug, loss_ctr_q and loss_ctr_U was calculated on 7 batch tasks)