a

1 files changed, 6 insertions, 0 deletions

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':