Skip to content
Snippets Groups Projects
Commit 24808e5a authored by Sun Jin Kim's avatar Sun Jin Kim
Browse files

revert work on evo_learner

parent 65eee061
No related branches found
No related tags found
No related merge requests found
Pipeline #272065 passed
from cgi import test
import torch import torch
import torch.nn as nn import torch.nn as nn
import pygad import pygad
...@@ -31,15 +30,18 @@ class evo_learner(aa_learner): ...@@ -31,15 +30,18 @@ class evo_learner(aa_learner):
controller=cont_n.evo_controller controller=cont_n.evo_controller
): ):
super().__init__(sp_num, super().__init__(
fun_num, sp_num=sp_num,
p_bins, p_bins=p_bins,
m_bins, m_bins=m_bins,
batch_size=batch_size, discrete_p_m=discrete_p_m,
toy_size=toy_size, batch_size=batch_size,
learning_rate=learning_rate, toy_size=toy_size,
max_epochs=max_epochs, learning_rate=learning_rate,
early_stop_num=early_stop_num,) max_epochs=max_epochs,
early_stop_num=early_stop_num,
exclude_method=exclude_method
)
# evolutionary algorithm settings # evolutionary algorithm settings
self.controller = controller( self.controller = controller(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment