From 0d5d0b4f381cab3478148640954692a3fc08afc8 Mon Sep 17 00:00:00 2001
From: Krishna Krishna Nikhil <ep21btech11016@iith.ac.in>
Date: Fri, 26 Jul 2024 08:21:47 +0000
Subject: [PATCH] Update optimizer.py
---
modules/optimizer.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/optimizer.py b/modules/optimizer.py
index e217c0c..e204df3 100644
--- a/modules/optimizer.py
+++ b/modules/optimizer.py
@@ -97,10 +97,10 @@ class Optimizer(object):
def sigmoid(number):
return 1/ (1 + torch.exp(-number))
- abs_sigm = sigmoid(scale_factor*(abs_parameters))
- cost_abs = abs_sigm * 25 + (1 - abs_sigm) * 4.166
+ abs_sigm = sigmoid(scale_factor*(abs_parameters)) # Scaled sigmoid function
+ cost_abs = abs_sigm * 25 + (1 - abs_sigm) * 4.166 # 25 is the cost of Pb and 4.166 for Fe
scint_sigm = sigmoid(scale_factor*(scint_parameters))
- cost_scint = scint_sigm * 2500 + (1 - scint_sigm) * 0.01
+ cost_scint = scint_sigm * 2500 + (1 - scint_sigm) * 0.01 # 2500 is the cost of PbWO4 and 0.01 for Polystyrene.
combined = torch.cat((cost_abs, cost_scint), dim=0)
--
GitLab