update

app.py

@@ -166,25 +166,25 @@ def main(args):
     # TODO, how to set the scale?
     logger.log("create dataset")
 
-    if args.objv_dataset:
-        from datasets.g_buffer_objaverse import load_data, load_eval_data, load_memory_data, load_wds_data
-    else:  # shapenet
-        from datasets.shapenet import load_data, load_eval_data, load_memory_data
+    # if args.objv_dataset:
+    #     from datasets.g_buffer_objaverse import load_data, load_eval_data, load_memory_data, load_wds_data
+    # else:  # shapenet
+    #     from datasets.shapenet import load_data, load_eval_data, load_memory_data
     
     # load data if i23d
-    if args.i23d:
-        data = load_eval_data(
-            file_path=args.eval_data_dir,
-            batch_size=args.eval_batch_size,
-            reso=args.image_size,
-            reso_encoder=args.image_size_encoder,  # 224 -> 128
-            num_workers=args.num_workers,
-            load_depth=True,  # for evaluation
-            preprocess=auto_encoder.preprocess,
-            **args_to_dict(args,
-                            dataset_defaults().keys()))
-    else:
-        data = None # t23d sampling, only caption required
+    # if args.i23d:
+    #     data = load_eval_data(
+    #         file_path=args.eval_data_dir,
+    #         batch_size=args.eval_batch_size,
+    #         reso=args.image_size,
+    #         reso_encoder=args.image_size_encoder,  # 224 -> 128
+    #         num_workers=args.num_workers,
+    #         load_depth=True,  # for evaluation
+    #         preprocess=auto_encoder.preprocess,
+    #         **args_to_dict(args,
+    #                         dataset_defaults().keys()))
+    # else:
+    data = None # t23d sampling, only caption required
 
 
     TrainLoop = {