|
|
|
|
|
|
super().__init__() |
|
|
|
self.normalize = network_settings.normalize |
|
|
|
self.use_lstm = network_settings.memory is not None |
|
|
|
self.h_size = network_settings.hidden_units |
|
|
|
# Scale network depending on num agents |
|
|
|
self.h_size = network_settings.hidden_units * num_obs_heads |
|
|
|
self.m_size = ( |
|
|
|
network_settings.memory.memory_size |
|
|
|
if network_settings.memory is not None |
|
|
|