fixes
This commit is contained in:
32
.ckpt_count.py
Normal file
32
.ckpt_count.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import sys, json, os, traceback
|
||||
sys.path.insert(0, r'F:\projects\gogo2')
|
||||
res={}
|
||||
try:
|
||||
from utils.database_manager import get_database_manager
|
||||
db=get_database_manager()
|
||||
def db_count(name):
|
||||
try:
|
||||
lst = db.list_checkpoints(name)
|
||||
return len(lst) if lst is not None else 0
|
||||
except Exception as e:
|
||||
print("DB error for %s: %s" % (name, str(e)))
|
||||
return -1
|
||||
res.setdefault('db', {})['dqn_agent']=db_count('dqn_agent')
|
||||
res['db']['enhanced_cnn']=db_count('enhanced_cnn')
|
||||
except Exception as e:
|
||||
res['db']={'error': str(e)}
|
||||
try:
|
||||
from utils.checkpoint_manager import get_checkpoint_manager
|
||||
cm=get_checkpoint_manager()
|
||||
def fs_count(name):
|
||||
try:
|
||||
lst = cm.get_all_checkpoints(name)
|
||||
return len(lst) if lst is not None else 0
|
||||
except Exception as e:
|
||||
print("FS error for %s: %s" % (name, str(e)))
|
||||
return -1
|
||||
res.setdefault('fs', {})['dqn_agent']=fs_count('dqn_agent')
|
||||
res['fs']['enhanced_cnn']=fs_count('enhanced_cnn')
|
||||
except Exception as e:
|
||||
res['fs']={'error': str(e)}
|
||||
print(json.dumps(res))
|
||||
Reference in New Issue
Block a user