From aa2523a9a98136dabad057a1ed890688bc9355b1 Mon Sep 17 00:00:00 2001
From: Yandi <yandirzm@gmail.com>
Date: Sat, 21 Jan 2023 20:51:58 +0100
Subject: [PATCH] [Run] Python script for optimizers
---
models/choose_optimizer.py | 5 +++++
1 file changed, 5 insertions(+)
create mode 100644 models/choose_optimizer.py
diff --git a/models/choose_optimizer.py b/models/choose_optimizer.py
new file mode 100644
index 0000000..1504fa7
--- /dev/null
+++ b/models/choose_optimizer.py
@@ -0,0 +1,5 @@
+import torch.optim
+
+def optimizer(cfg):
+ result = {"Adam" : torch.optim.Adam(model.parameters())}
+ return result[cfg["Optimizer"]]
--
GitLab