import os
import numpy as np
import glob
import time
import datetime
import sys

l_folders = np.sort(glob.glob("/nfs/NAS6/SABIOD/SITE/CARIMAM/DATA/LOT1[0-9]/*"))
idx_gpu = int(sys.argv[1])

for folder in l_folders :
  print("Processing: %s"%folder)
  t_beg = time.time()
  os.system("python3 forward_both.py '%s' --path_out result_both/ --weight context_carimam/05-17-23_17\:29\:22\:cstm_name\=TRANS_1024_MAXPOOL\:hp\=True\:lr\=0.001\:ne\=2:wd\=0.05\:bs\=8:lmbd\=0.9:blcd\=False\:aug\=True\:frame\=1024/ckpt_5632.pth --nb_frame 1024 --bs 8 --hp --nb_workers 12 --device %d --vit"%(folder, idx_gpu))

  t_end = time.time()
  print("End in : %s"%(datetime.timedelta(seconds=int(t_end-t_beg))))