cam_ext_mask_prob is use only for the first camera

[vaxowt]

In line 299, only the first element of mask_probs is set to cam_ext_mask_prob

RoboticsDiffusionTransformer/train/dataset.py

Lines 297 to 309 in 49866c7

	mask_probs = [self.cond_mask_prob] * self.num_cameras
	if self.cam_ext_mask_prob >= 0.0:
	mask_probs[0] = self.cam_ext_mask_prob
	rearranged_images = []
	for i in range(self.img_history_size):
	for j in range(self.num_cameras):
	images, image_mask = image_metas[j]
	image, valid = images[i], image_mask[i]
	if valid and (math.prod(image.shape) > 0) and \
	(random.random() > mask_probs[j]):
	rearranged_images.append((image, True))
	else:
	rearranged_images.append((background_image.copy(), False))