Index A | B | C | D | E | F | G | H | I | J | L | M | N | P | R | S | T | U | V | W A add() (PointSequential method) adjust_brightness() (RandomColorJitter method) adjust_contrast() (RandomColorJitter method) adjust_hue() (RandomColorJitter method) adjust_saturation() (RandomColorJitter method) after_epoch() (CheckpointSaver method) (CheckpointSaverIteration method) (HookBase method) (InformationWriter method) (InstanceSegmentationEvaluator method) (LogitEntropyLogger method) (MAEEvaluator method) (ModelHook method) (PretrainEvaluator method) (PushToHub method) (SemSegEvaluator method) (Trainer method) after_step() (AttentionMaskAnnealingHook method) (CheckpointSaver method) (CheckpointSaverIteration method) (GarbageHandler method) (GradientNormLogger method) (HookBase method) (InformationWriter method) (InstanceSegmentationEvaluator method) (IterationTimer method) (LogitEntropyLogger method) (MAEEvaluator method) (ModelHook method) (PretrainEvaluator method) (PushToHub method) (ResourceUtilizationLogger method) (SemSegEvaluator method) (SimulateCrash method) (Sonata method), [1] after_train() (CheckpointSaver method) (CheckpointSaverIteration method) (FeatureStdMonitor method) (FinalEvaluator method) (GarbageHandler method) (HookBase method) (ModelHook method) (PrototypeUsageLogger method) (PushToHub method) (ResourceUtilizationLogger method) (SemSegEvaluator method) all_gather() (in module pimm.utils.comm) AttentionMaskAnnealingHook (class in pimm.engines.hooks.diagnostics) auto_argparser() (Config static method) B batch_size_per_rank (TrainState attribute) before_epoch() (GarbageHandler method) (HookBase method) (IterationTimer method) (ModelHook method) before_step() (HookBase method) (InformationWriter method) (IterationTimer method) (ModelHook method) (Sonata method), [1] (WeightDecayScheduler method) before_train() (AttentionMaskAnnealingHook method) (CheckpointLoader method) (CheckpointSaver method) (CheckpointSaverIteration method) (DtypeOverrider method) (FeatureStdMonitor method) (GarbageHandler method) (HookBase method) (InformationWriter method) (IterationTimer method) (LogitEntropyLogger method) (ModelHook method) (ParameterCounter method) (PrototypeUsageLogger method) (PushToHub method) (ResourceUtilizationLogger method) (RuntimeProfiler method) (Sonata method), [1] (WeightDecayExclusion method) (WeightDecayScheduler method) best_metric_value (TrainState attribute) BinaryFocalLoss (class in pimm.models.losses.misc) blend() (RandomColorJitter static method) BLOCK (MinkUNet101 attribute) (MinkUNet14 attribute) (MinkUNet18 attribute) (MinkUNet34 attribute) (MinkUNet50 attribute) build() (Registry method) build_criteria() (in module pimm.models.losses.builder) build_dataset() (in module pimm.datasets.builder) build_model() (in module pimm.models.builder) (Trainer method) build_optimizer() (Trainer method) build_scaler() (Trainer method) build_scheduler() (GRPOTrainer method) (Trainer method) build_test_loader() (Trainer method) build_train_loader() (ImageClassTrainer method) (InsegTrainer method) (MultiDatasetTrainer method) (Trainer method) build_val_loader() (ImageClassTrainer method) (InsegTrainer method) (Trainer method) build_writer() (Trainer method) C CenterShift (class in pimm.datasets.transform.spatial) CheckpointLoader (class in pimm.engines.hooks.checkpoint) CheckpointManager (class in pimm.utils.checkpoints) CheckpointSaver (class in pimm.engines.hooks.checkpoint) CheckpointSaverIteration (class in pimm.engines.hooks.checkpoint) children (Registry property) ChromaticAutoContrast (class in pimm.datasets.transform.color) ChromaticJitter (class in pimm.datasets.transform.color) ChromaticTranslation (class in pimm.datasets.transform.color) clean_state_dict() (in module pimm.export) ClipGaussianJitter (class in pimm.datasets.transform.spatial) collate_fn() (in module pimm.datasets.utils) Collect (class in pimm.datasets.transform.base) Compose (class in pimm.datasets.transform.base) ComputeAnchors (class in pimm.datasets.transform.instance) ConcatDataset (class in pimm.datasets.defaults) ConditionalRandomTransform (class in pimm.datasets.transform.spatial) Config (class in pimm.utils.config) ContrastiveViewsGenerator (class in pimm.datasets.transform.multiview) Copy (class in pimm.datasets.transform.base) Criteria (class in pimm.models.losses.builder) CropBoundary (class in pimm.datasets.transform.spatial) CrossEntropyHeadLoss (class in pimm.models.losses.misc) CrossEntropyLoss (class in pimm.models.losses.misc) D dataloader_state (TrainState attribute) default_setup() (in module pimm.engines.defaults) DefaultClassifier (class in pimm.models.default) DefaultDataset (class in pimm.datasets.defaults) DefaultInsSegmentor (class in pimm.models.default) DefaultSegmentor (class in pimm.models.default) DefaultSegmentorV2 (class in pimm.models.default) DefaultSegmentorV3 (class in pimm.models.default) deprecated_register_module() (Registry method) DiceLoss (class in pimm.models.losses.misc) DINOEnhancedSegmentor (class in pimm.models.default) DtypeOverrider (class in pimm.engines.hooks.diagnostics) dump() (Config method) E elastic_distortion() (ElasticDistortion static method) ElasticDistortion (class in pimm.datasets.transform.spatial) EnergeticTranslation (class in pimm.datasets.transform.color) EnergyJitter (class in pimm.datasets.transform.color) epoch (TrainState attribute) eval() (InstanceSegmentationEvaluator method) (MAEEvaluator method) (PretrainEvaluator method) (SemSegEvaluator method) eval_per_instance() (SemSegEvaluator method) extra_state (TrainState attribute) F FastInstanceSegmentationLoss (class in pimm.models.losses.instance_fast) FastInstanceSegmentationRegressionLoss (class in pimm.models.losses.instance_regression_fast) FastUnifiedInstanceLoss (class in pimm.models.losses.instance_unified_fast) FeatureStdMonitor (class in pimm.engines.hooks.diagnostics) filename (Config property) filter_state_dict_by_prefix() (in module pimm.export) FinalEvaluator (class in pimm.engines.hooks.eval.final_tester) fnv_hash_vec() (GridSample static method) FocalLoss (class in pimm.models.losses.misc) forward() (BinaryFocalLoss method) (CrossEntropyHeadLoss method) (CrossEntropyLoss method) (DefaultClassifier method) (DefaultInsSegmentor method) (DefaultSegmentor method) (DefaultSegmentorV2 method) (DefaultSegmentorV3 method) (DiceLoss method) (DINOEnhancedSegmentor method) (FastInstanceSegmentationLoss method) (FastInstanceSegmentationRegressionLoss method) (FastUnifiedInstanceLoss method) (FocalLoss method) (L1RegressionLoss method) (LitePT method) (LoRAAdapter method) (LovaszLoss method) (MSERegressionLoss method) (MultiLabelDetectorV3M2 method) (PointGroup method) (PointSequential method) (PointTransformerV2 method), [1], [2] (PointTransformerV3 method), [1], [2] (PoLArMAE method) (PoLArMAESemSeg method) (SmoothCELoss method) (SmoothL1RegressionLoss method) (Sonata method), [1] (SpUNetBase method), [1], [2] (SpUNetNoSkipBase method) (UnifiedDetector method), [1], [2] (VoltBackbone method) (VoltBackboneV2 method) from_pretrained() (in module pimm) from_state_dict() (TrainState class method) from_trainer() (TrainState class method) fromfile() (Config static method) fromstring() (Config static method) fuse_representation() (Sonata method) G GarbageHandler (class in pimm.engines.hooks.resources) generate_mask() (Sonata method), [1] get() (Registry method) get_backbone() (PretrainEvaluator method) get_center() (MultiViewGenerator method) get_data() (ConcatDataset method) (DefaultDataset method) (JAXTPCDataset method) (LUCiDDataset method) (PILArNetH5Dataset method) (PILArNetParquetDataset method) get_data_list() (ConcatDataset method) (DefaultDataset method) get_data_name() (ConcatDataset method) (DefaultDataset method) (JAXTPCDataset method) (LUCiDDataset method) (PILArNetH5Dataset method) (PILArNetParquetDataset method) get_h5_files() (PILArNetH5Dataset method) get_params() (RandomColorJitter static method) get_rank() (in module pimm.utils.comm) get_split_name() (DefaultDataset method) get_view() (MultiViewGenerator method) get_world_size() (in module pimm.utils.comm) global_step (TrainState attribute) grad_accum_step (TrainState attribute) GradientNormLogger (class in pimm.engines.hooks.diagnostics) GridSample (class in pimm.datasets.transform.spatial) GRPOTrainer (class in pimm.engines.train) H h5py_worker_init() (PILArNetH5Dataset method) HardExampleCrop (class in pimm.datasets.transform.spatial) HookBase (class in pimm.engines.hooks.default) hsv2rgb() (RandomColorJitter static method) hsv_to_rgb() (HueSaturationTranslation static method) HueSaturationTranslation (class in pimm.datasets.transform.color) I ImageClassTrainer (class in pimm.engines.train) infer_scope() (Registry static method) InformationWriter (class in pimm.engines.hooks.logging) InsegTrainer (class in pimm.engines.train) InstanceParser (class in pimm.datasets.transform.instance) InstanceSegmentationEvaluator (class in pimm.engines.hooks.eval.instance_segmentation) InstanceSegmentationLoss (class in pimm.models.losses.instance_fast) is_main_process() (in module pimm.utils.comm) iter_in_epoch (TrainState attribute) iteration (TrainState property) IterationTimer (class in pimm.engines.hooks.logging) J JAXTPCDataset (class in pimm.datasets.jaxtpc_dataset) L L1RegressionLoss (class in pimm.models.losses.misc) LAYERS (MinkUNet101 attribute) (MinkUNet14 attribute) (MinkUNet18 attribute) (MinkUNet34 attribute) (MinkUNet50 attribute) linear_transform() (LogTransform method) LitePT (class in pimm.models.litept.litept) load_checkpoint_metadata() (in module pimm.export) load_optimizer_state() (CheckpointManager method) load_pretrained() (in module pimm.export) load_state_dict() (StatefulRandomSampler method) (TrainState method) load_state_dict_from_checkpoint() (in module pimm.export) load_weight_and_resume() (CheckpointManager method) LocalCovarianceFeatures (class in pimm.datasets.transform.instance) log_transform() (LogTransform method) LogitEntropyLogger (class in pimm.engines.hooks.diagnostics) LogTransform (class in pimm.datasets.transform.color) LoRAAdapter (class in pimm.models.lora) LOSSES (in module pimm.models.losses.builder) LovaszLoss (class in pimm.models.losses.lovasz) LUCiDDataset (class in pimm.datasets.lucid_dataset) M MAEEvaluator (class in pimm.engines.hooks.eval.pretrain.mae) match_neighbour() (Sonata method), [1] merge_from_dict() (Config method) MinkUNet101 (class in pimm.models.sparse_unet.mink_unet) MinkUNet14 (class in pimm.models.sparse_unet.mink_unet) MinkUNet14A (class in pimm.models.sparse_unet.mink_unet) MinkUNet14B (class in pimm.models.sparse_unet.mink_unet) MinkUNet14C (class in pimm.models.sparse_unet.mink_unet) MinkUNet14D (class in pimm.models.sparse_unet.mink_unet) MinkUNet18 (class in pimm.models.sparse_unet.mink_unet) MinkUNet18A (class in pimm.models.sparse_unet.mink_unet) MinkUNet18B (class in pimm.models.sparse_unet.mink_unet) MinkUNet18D (class in pimm.models.sparse_unet.mink_unet) MinkUNet34 (class in pimm.models.sparse_unet.mink_unet) MinkUNet34A (class in pimm.models.sparse_unet.mink_unet) MinkUNet34B (class in pimm.models.sparse_unet.mink_unet) MinkUNet34C (class in pimm.models.sparse_unet.mink_unet) MinkUNet50 (class in pimm.models.sparse_unet.mink_unet) MixedScaleGeometryMultiViewGenerator (class in pimm.datasets.transform.multiview) ModelHook (class in pimm.engines.hooks.default) MODELS (in module pimm.models.builder) modify_config() (HookBase method) (WandbNamer method) module_dict (Registry property) MomentumTransform (class in pimm.datasets.transform.color) MSERegressionLoss (class in pimm.models.losses.misc) MultiDatasetTrainer (class in pimm.engines.train) MultiLabelDetectorV3M2 (class in pimm.models.panda_detector.detector_v3m2) MultiplicativeRandomJitter (class in pimm.datasets.transform.spatial) MultiViewGenerator (class in pimm.datasets.transform.multiview) N name (Registry property) NormalizeColor (class in pimm.datasets.transform.color) NormalizeCoord (class in pimm.datasets.transform.spatial) num_samples() (PILArNetParquetIterableDataset method) P packed_to_batched() (in module pimm.models.polarmae.data) ParameterCounter (class in pimm.engines.hooks.diagnostics) PDGToSemantic (class in pimm.datasets.transform.detector) PILArNetH5Dataset (class in pimm.datasets.pilarnet.h5) PILArNetParquetDataset (class in pimm.datasets.pilarnet.parquet) PILArNetParquetIterableDataset (class in pimm.datasets.pilarnet.parquet) PLANES (MinkUNet14A attribute) (MinkUNet14B attribute) (MinkUNet14C attribute) (MinkUNet14D attribute) (MinkUNet18A attribute) (MinkUNet18B attribute) (MinkUNet18D attribute) (MinkUNet34A attribute) (MinkUNet34B attribute) (MinkUNet34C attribute) Point (class in pimm.models.utils.structure) PointClip (class in pimm.datasets.transform.spatial) PointGroup (class in pimm.models.point_group.point_group_v1m1_base) PointModel (class in pimm.models.modules) PointModule (class in pimm.models.modules) PointSequential (class in pimm.models.modules) PointTransformerCls26 (class in pimm.models.point_transformer.point_transformer_cls) PointTransformerCls38 (class in pimm.models.point_transformer.point_transformer_cls) PointTransformerCls50 (class in pimm.models.point_transformer.point_transformer_cls) PointTransformerSeg26 (class in pimm.models.point_transformer.point_transformer_partseg) (class in pimm.models.point_transformer.point_transformer_seg) PointTransformerSeg38 (class in pimm.models.point_transformer.point_transformer_partseg) (class in pimm.models.point_transformer.point_transformer_seg) PointTransformerSeg50 (class in pimm.models.point_transformer.point_transformer_partseg) (class in pimm.models.point_transformer.point_transformer_seg) PointTransformerV2 (class in pimm.models.point_transformer_v2.point_transformer_v2m1_origin) (class in pimm.models.point_transformer_v2.point_transformer_v2m2_base) (class in pimm.models.point_transformer_v2.point_transformer_v2m3_pdnorm) PointTransformerV3 (class in pimm.models.point_transformer_v3.point_transformer_v3m1_base) (class in pimm.models.point_transformer_v3.point_transformer_v3m2_sonata) (class in pimm.models.point_transformer_v3.point_transformer_v3m8) PoLArMAE (class in pimm.models.polarmae.polarmae) PoLArMAESemSeg (class in pimm.models.polarmae.polarmae_semseg) PositiveShift (class in pimm.datasets.transform.spatial) postprocess() (MultiLabelDetectorV3M2 method) (UnifiedDetector method), [1], [2] prepare_test_data() (DefaultDataset method) (JAXTPCDataset method) (LUCiDDataset method) (PILArNetH5Dataset method) prepare_train_data() (DefaultDataset method) (JAXTPCDataset method) (LUCiDDataset method) (PILArNetH5Dataset method) PretrainEvaluator (class in pimm.engines.hooks.eval.pretrain.semantic_segmentation_pretrain) pretty_text (Config property) PrototypeUsageLogger (class in pimm.engines.hooks.diagnostics) push_to_hub() (in module pimm) PushToHub (class in pimm.engines.hooks.export) R RandomColorDrop (class in pimm.datasets.transform.color) RandomColorGrayScale (class in pimm.datasets.transform.color) RandomColorJitter (class in pimm.datasets.transform.color) RandomDrop (class in pimm.datasets.transform.spatial) RandomDropout (class in pimm.datasets.transform.spatial) RandomFlip (class in pimm.datasets.transform.spatial) RandomJitter (class in pimm.datasets.transform.spatial) RandomRotate (class in pimm.datasets.transform.spatial) RandomRotateTargetAngle (class in pimm.datasets.transform.spatial) RandomScale (class in pimm.datasets.transform.spatial) RandomShift (class in pimm.datasets.transform.spatial) ravel_hash_vec() (GridSample static method) reduce_dict() (in module pimm.utils.comm) register_module() (Registry method) Registry (class in pimm.utils.registry) relative_log_transform() (RelativeLogNormalize method) RelativeLogNormalize (class in pimm.datasets.transform.color) remaining (StatefulRandomSampler property) remap_state_dict_keys() (in module pimm.export) ResourceUtilizationLogger (class in pimm.engines.hooks.resources) resume_training_state() (CheckpointManager method) rgb2hsv() (RandomColorJitter static method) rgb_to_grayscale() (RandomColorGrayScale static method) rgb_to_hsv() (HueSaturationTranslation static method) rng_state (TrainState attribute) roll_point() (Sonata method), [1] run_step() (GRPOTrainer method) (Trainer method) RuntimeProfiler (class in pimm.engines.hooks.profiling) S sampler_state (TrainState attribute) samples_seen (TrainState attribute) save_epoch_checkpoint() (CheckpointManager method) save_iteration_checkpoint() (CheckpointManager method) save_pretrained() (in module pimm) schema_version (TrainState attribute) scope (Registry property) SemSegEvaluator (class in pimm.engines.hooks.eval.semantic_segmentation) serialization() (Point method) set_epoch() (PILArNetParquetIterableDataset method) (StatefulRandomSampler method) set_position() (StatefulRandomSampler method) SetRandomValue (class in pimm.datasets.transform.spatial) ShufflePoint (class in pimm.datasets.transform.spatial) SimulateCrash (class in pimm.engines.hooks.diagnostics) sinkhorn_knopp() (Sonata static method), [1] SmoothCELoss (class in pimm.models.losses.misc) SmoothL1RegressionLoss (class in pimm.models.losses.misc) Sonata (class in pimm.models.sonata.sonata_v1m1_base) (class in pimm.models.sonata.sonata_v1m2_uni_teacher_head) sparsify() (Point method) SphereCrop (class in pimm.datasets.transform.spatial) split_scope_key() (Registry static method) SpUNetBase (class in pimm.models.sparse_unet.spconv_unet_v1m1_base) (class in pimm.models.sparse_unet.spconv_unet_v1m2_bn_momentum) (class in pimm.models.sparse_unet.spconv_unet_v1m3_pdnorm) SpUNetNoSkipBase (class in pimm.models.sparse_unet.spconv_unet_v1m1_base) state_dict() (StatefulRandomSampler method) (TrainState method) StatefulRandomSampler (class in pimm.datasets.stateful) synchronize() (in module pimm.utils.comm) T text (Config property) ToTensor (class in pimm.datasets.transform.base) train() (DefaultSegmentorV2 method) (Trainer method) Trainer (class in pimm.engines.train) trainer (HookBase attribute) TrainState (class in pimm.engines._train_utils) U UnifiedDetector (class in pimm.models.panda_detector.detector_v4) (class in pimm.models.panda_detector.detector_v5) (class in pimm.models.panda_detector.detector_v5m2) up_cast() (MultiLabelDetectorV3M2 method) (Sonata method), [1] (UnifiedDetector method), [1], [2] Update (class in pimm.datasets.transform.base) update_anneal_step() (MultiLabelDetectorV3M2 method) (UnifiedDetector method), [1], [2] upsample_to_original() (Sonata method) V VALID_ASSETS (DefaultDataset attribute) VoltBackbone (class in pimm.models.volt.volt_v1m1) VoltBackboneV2 (class in pimm.models.volt.volt_v1m2) W WandbNamer (class in pimm.engines.hooks.logging) WeightDecayExclusion (class in pimm.engines.hooks.optimizer) WeightDecayScheduler (class in pimm.engines.hooks.optimizer) world_size (TrainState attribute)
