TypeError: TriPlaneSemanticEntangleGenerator.forward() missing 1 required positional argument: 'batch'

[beijing-penguin]

python train.py --outdir="./my_train_out" --cfg=afhq --data=data/afhq_v2_train_cat_512.zip --mask_data=data/afhqcat_seg_6c_v2.zip --data_type=seg --semantic_channels=6 --render_mask=True --dis_mask=True --neural_rendering_resolution_initial=64 --gpus=1 --batch=1 --mbstd-group=1 --gamma=5 --gen_pose_cond=True --random_c_prob=0.5 --lambda_d_semantic=0.1 --lambda_lpips=1 --lambda_cross_view=1e-4 --only_raw_recons=True --wandb_log=False --no_eval=False

when i use --no_eval=False ,Code throw exception,as follow:

  File "E:\python-eclipse-workspace-2022-12\pix2pix3D\metrics\metric_main.py", line 96, in kid50k_full
    kid = kernel_inception_distance.compute_kid(opts, max_real=1000000, num_gen=50000, num_subsets=100, max_subset_size=1000)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\python-eclipse-workspace-2022-12\pix2pix3D\metrics\kernel_inception_distance.py", line 29, in compute_kid
    gen_features = metric_utils.compute_feature_stats_for_generator(
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\python-eclipse-workspace-2022-12\pix2pix3D\metrics\metric_utils.py", line 287, in compute_feature_stats_for_generator
    img = G(z=z, c=it_data, **opts.G_kwargs)['image']
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Python\Python3_11_3\Lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: TriPlaneSemanticEntangleGenerator.forward() missing 1 required positional argument: 'batch'

the G is TriPlaneSemanticEntangleGenerator