-
Notifications
You must be signed in to change notification settings - Fork 2
[DRAFT] BC + ResNet + LSTM #5
Conversation
algoritms/bc_resnet_lstm.py
Outdated
|
||
states, actions, *_ = batch | ||
logits, rnn_state = actor( | ||
states.permute(0, 1, 4, 2, 3).to(DEVICE).to(torch.float32), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
мб лучше einops?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
хз, я если честно его не очень понимаю, но если все за, то мб
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
у него просто более интуитивная форма, тут получается было бы "b t x y c -> b t c x y", что более интуитивно выглядит
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
то есть, понимать, что у эйнопса под коробкой происходит необязательно, чтобы input->output семантику улавливать
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
а он точно не дает оверхеда?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
не уверен, надо чекать
No description provided.