Configuration of latent action model. #3
Description
There is no config file for training the latent action model.
Also, in the example code of training the latent action model, there is no "n_embed" for space-time_attn:
('space-time_attn', { 'n_rep' : 2, 'n_embd' : 256, 'n_head' : 4, 'd_head' : 16, 'has_ext' : True, # Decoder uses latent action as external # conditioning for decoding! 'time_attn_kw' : {'key_dim' : 8}, })