mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01:00
network_imporvements branch merge
This commit is contained in:
@@ -46,8 +46,6 @@ class HumanNetworkParameters(NetworkParameters):
|
||||
self.input_embedders_parameters = {'observation': InputEmbedderParameters()}
|
||||
self.input_embedders_parameters['observation'].scheme = EmbedderScheme.Medium
|
||||
self.middleware_parameters = FCMiddlewareParameters()
|
||||
self.heads_parameters = [PolicyHeadParameters()]
|
||||
self.loss_weights = [1.0]
|
||||
self.optimizer_type = 'Adam'
|
||||
self.batch_size = 32
|
||||
self.replace_mse_with_huber_loss = False
|
||||
|
||||
Reference in New Issue
Block a user