cpu pls

test_few_shot.py

@@ -23,7 +23,7 @@ def test_main_model(opts):
         dir_res = os.path.join(f"{opts.exp_path}", "experiments/", opts.name_exp, "results")
 
     test_loader = get_loader(opts.data_root, opts.img_size, opts.language, opts.char_num, opts.max_seq_len, opts.dim_seq, opts.batch_size, 'test')
-    
+
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     print("Inference With Device:", device)
     if opts.streamlit:
@@ -75,8 +75,8 @@ def test_main_model(opts):
                     st.progress((sample_idx+1)/opts.n_samples, f"Generating Font Sample {sample_idx+1} Please wait...")
                     im = Image.open(save_file_merge)
                     st.image(im, caption='img_sample_merge')
-
-                for char_idx in range(opts.char_num):
+                
+                for char_idx in tqdm(range(opts.char_num)):
                     img_gt = (1.0 - img_trg[char_idx,...]).data
                     save_file_gt = os.path.join(dir_save,"imgs", f"{char_idx:02d}_gt.png")
                     save_image(img_gt, save_file_gt, normalize=True)
@@ -87,7 +87,7 @@ def test_main_model(opts):
 
                 # write results w/o parallel refinement
                 svg_dec_out = svg_sampled.clone().detach()
-                for i, one_seq in enumerate(svg_dec_out):
+                for i, one_seq in tqdm(enumerate(svg_dec_out)):
                     syn_svg_outfile = os.path.join(os.path.join(dir_save, "svgs_single"), f"syn_{i:02d}_{sample_idx}_wo_refine.svg")
 
                     syn_svg_f_ = open(syn_svg_outfile, 'w')
@@ -105,7 +105,7 @@ def test_main_model(opts):
                 
                 # write results w/ parallel refinement
                 svg_dec_out = sampled_svg_2.clone().detach()
-                for i, one_seq in enumerate(svg_dec_out):
+                for i, one_seq in tqdm(enumerate(svg_dec_out)):
                     syn_svg_outfile = os.path.join(os.path.join(dir_save, "svgs_single"), f"syn_{i:02d}_{sample_idx}_refined.svg")
 
                     syn_svg_f = open(syn_svg_outfile, 'w')
@@ -127,7 +127,7 @@ def test_main_model(opts):
                         iou_max[i] = iou_tmp
                         idx_best_sample[i] = sample_idx
 
-            for i in range(opts.char_num):
+            for i in tqdm(range(opts.char_num)):
                 # print(idx_best_sample[i])
                 syn_svg_outfile_best = os.path.join(os.path.join(dir_save, "svgs_single"), f"syn_{i:02d}_{int(idx_best_sample[i])}_refined.svg")
                 syn_svg_merge_f.write(open(syn_svg_outfile_best, 'r').read())