RLE-Foundation / rllte

Long-Term Evolution Project of Reinforcement Learning

https://docs.rllte.dev/

MIT License

453 stars 84 forks source link

Added ENet, Transformer and Mobilenet to encoder #4

Closed WilliamYangXu closed 1 year ago

WilliamYangXu commented 1 year ago

encoder folder update

Added the following files to Hsuanwu/hsuanwu/xploit/encoder:

enet_encoder.py
transformer_encoder.py
mobilenet_encoder.py

In the current version, mobilenet has one block, enet contains three blocks with user-specified number of input channels, and transformer consists of arbitrary blocks specified by user. Will make updates.