search

stepjam / TecNets

Official code for "Task-Embedded Control Networks for Few-Shot Imitation Learning".
Other
44 stars 10 forks source link

Why only one timestep for the imitation loss? #5

Open telejesus2 opened 3 years ago

telejesus2 commented 3 years ago

Hi! I have noticed that during training the control network has to predict only one action per episode (one from a query demo and one from a support demo so two in total).

I am curious what is the goal of this? Is it a way to balance the training of the embedding and control networks? Did you try other strategies (e.g. predicting all actions of all demos in the episode, or something in between)?

Thanks!