메모리 누수 해결 - 완전한 해결은 아니라 대량의 메모리가 필요
mnist 최적값을 찾는 파티클 개수 찾아야 함
This commit is contained in:
jung-geun
2023-07-23 18:37:20 +09:00
parent 7d1161aa95
commit f692ff7b4a
6 changed files with 33 additions and 649 deletions

View File

@@ -106,11 +106,11 @@ loss = [
pso_mnist = Optimizer(
model,
loss=loss[0],
n_particles=100,
c0=0.25,
c1=0.4,
w_min=0.3,
w_max=0.9,
n_particles=1000,
c0=0.4,
c1=0.6,
w_min=0.5,
w_max=0.8,
negative_swarm=0.1,
mutation_swarm=0.2,
particle_min=-5,