jung-geun/PSO
1
0
Fork 0
mirror of https://github.com/jung-geun/PSO.git synced 2025-12-20 04:50:45 +09:00
Code Issues Packages Projects Releases Wiki Activity

23-06-01

Browse Source 
함수 실행마다 사용안하는 변수 delete 및 gc.collect() 를 실행하여 메모리 문제 해결을 위해 변경
This commit is contained in:
jung-geun
2023-06-01 18:10:57 +09:00
parent 89449048c4
commit 4ffc6cc6e5
6 changed files with 164 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mnist.py
View File

@@ -79,9 +79,9 @@ loss = 'huber_loss'
# loss = 'mean_squared_error'
pso_mnist = Optimizer(model, loss=loss, n_particles=75, c0=0.4, c1=0.8, w_min=0.6, w_max=0.95, random=0.3)
pso_mnist = Optimizer(model, loss=loss, n_particles=50, c0=0.4, c1=0.8, w_min=0.4, w_max=0.95, negative_swarm=0.3)
weight, score = pso_mnist.fit(
x_test, y_test, epochs=500, save=True, save_path="./result/mnist", renewal="acc", empirical_balance=False, Dispersion=False, check_point=10)
x_test, y_test, epochs=500, save=True, save_path="./result/mnist", renewal="acc", empirical_balance=True, Dispersion=False, check_point=10)
# pso_mnist.model_save("./result/mnist")
# pso_mnist.save_info("./result/mnist")