glob load script

load_glob.py

import torch
import os
from glob import glob
from progressbar import ProgressBar

if __name__ == '__main__':
  proof_steps = glob(os.path.join('proof_steps_gnn', '**', '*.pt'), recursive=True)
  # print(proof_steps)
  bar = ProgressBar(max_value = len(proof_steps))
  for i, filename in enumerate(proof_steps):
    try:
      torch.load(filename)
     except:
      print(f'Exception in {filename}!')
     bar.update(i)