1029 add octuple

generate-batch.py

@@ -33,7 +33,9 @@ def get_argument_parser():
     parser.add_argument(
         "-attr_list",
         type=str,
-        default="beat,duration",
+        # default="beat,duration,,instrument,tempo",
+        default="pitch",
+        # default='bar,position,velocity,duration,program,tempo,timesig',
         help="attribute list for attribute-controlled generation",
     )
     parser.add_argument(
@@ -69,7 +71,7 @@ def get_argument_parser():
     parser.add_argument(
         "-num_target_measure",
         type=int,
-        default=4,
+        default=128,
         help="number of target measures for conditioned generation",
     )
     parser.add_argument(
@@ -86,13 +88,13 @@ def get_argument_parser():
     parser.add_argument(
         "-num_processes",
         type=int,
-        default=1,
+        default=4,
         help="number of processes to use",
     )
     parser.add_argument(
         "-gpu_ids",
         type=str,
-        default="1,2,3,5",
+        default="0,1,2,3,5",
         help="comma-separated list of GPU IDs to use (e.g., '0,1,2,3')",
     )
     parser.add_argument(
@@ -203,6 +205,7 @@ def attr_conditioned_worker(process_idx, gpu_id, args):
     # end_idx = min(chunk_size, len(selected_data))
     # data_slice = selected_data[start_idx:end_idx]
     data_slice = selected_data
+    print("data_slice length:", len(data_slice))
     
     # Create output directory with process index
     base_path = Path('wandb') / args.wandb_exp_dir / \