From ca53592e108e2189ef1b625c45d2b2a23b7ab145 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 30 Aug 2018 12:20:36 +0200 Subject: a --- rpc/listener.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/rpc/listener.py b/rpc/listener.py index 87eea37..81ad794 100644 --- a/rpc/listener.py +++ b/rpc/listener.py @@ -9,6 +9,10 @@ def list_checkpoints(payload): print("> list checkpoints") return sorted([f.split('/')[3] for f in glob.glob(os.path.join('./checkpoints/', payload, '/*/latest_net_G.pth'))]) +def list_all_checkpoints(payload): + print("> list all checkpoints") + return sorted(glob.glob(os.path.join('./checkpoints/*/*_net_G.pth'))) + def list_epochs(path): print("> list epochs for {}".format(path)) if not os.path.exists(os.path.join('./checkpoints/', path)): @@ -61,6 +65,8 @@ class Listener(): if cmd == 'list_checkpoints': return list_checkpoints(payload) + if cmd == 'list_all_checkpoints': + return list_all_checkpoints(payload) if cmd == 'list_epochs': return list_epochs(payload) if cmd == 'list_sequences': -- cgit v1.2.3-70-g09d2