23-09-04

#5 - 메모리의 점진적인 누수
파티클의 갯수 4000개 epochs 100회 기준 최종적으로 150GB 의 메모리 사용량을 보인다
하지만 초기의 메모리는 26GB 로 점진적으로 메모리의 사용량이 증가하는것으로 볼 수 있다

mnist.py

@@ -104,11 +104,11 @@ loss = [
 pso_mnist = optimizer(
     model,
     loss="mean_squared_error",
-    n_particles=990,
+    n_particles=2000,
     c0=0.2,
     c1=0.4,
     w_min=0.3,
-    w_max=0.6,
+    w_max=0.7,
     negative_swarm=0.1,
     mutation_swarm=0.3,
     particle_min=-4,
@@ -118,15 +118,16 @@ pso_mnist = optimizer(
 best_score = pso_mnist.fit(
     x_train,
     y_train,
-    epochs=200,
+    epochs=300,
     save_info=True,
-    log=2,
+    log=1,
     log_name="mnist",
-    save_path="./result/mnist",
+    save_path="./logs/mnist",
     renewal="acc",
     check_point=25,
     empirical_balance=False,
     dispersion=False,
+
 )
 
 print("Done!")