랜덤 시드를 42로 설정하여 난수 생성을 일관성 있게 변경

This commit is contained in:
jung-geun
2024-03-12 21:13:51 +09:00
parent ca44ff5f7a
commit 71cc5308f8
3 changed files with 16 additions and 7 deletions

View File

@@ -530,7 +530,7 @@ class Optimizer:
* (epoch % weight_reduction)
/ weight_reduction
)
rng = np.random.default_rng()
rng = np.random.default_rng(seed=42)
for i in part_pbar:
part_pbar.set_description(
f"loss: {min_loss:.4f} acc: {max_acc:.4f} mse: {min_mse:.4f}"