jgwill commited on
Commit
094d003
·
1 Parent(s): 305e0ff
Files changed (1) hide show
  1. compo-singleone-v2-dev-acc.py +1 -1
compo-singleone-v2-dev-acc.py CHANGED
@@ -100,7 +100,7 @@ def get_padded_checkpoint_no_from_filename(checkpoint_filename):
100
  print(checkpoint_number)
101
 
102
  padded_checkpoint_number = str(str(checkpoint_number).zfill(3))
103
- return padded_checkpoint_number
104
 
105
  found_model='none'
106
  found_model_checkpoint='0'
 
100
  print(checkpoint_number)
101
 
102
  padded_checkpoint_number = str(str(checkpoint_number).zfill(3))
103
+ return padded_checkpoint_number.replace('.0','')
104
 
105
  found_model='none'
106
  found_model_checkpoint='0'