MCPcopy Create free account

hub / github.com/PolymathicAI/AstroCLIP / functions

Functions197 in github.com/PolymathicAI/AstroCLIP

⨍ Functions 197◇ Types & classes 44
↓ 31 callersFunctionprint
(*args, **kwargs)
astroclip/astrodino/distributed.py:82
↓ 12 callersFunctionformat_with_env
(s: T)
astroclip/env.py:50
↓ 12 callersMethodupdate
(self, **kwargs)
astroclip/astrodino/utils.py:30
↓ 7 callersMethodsplit
(self)
astroclip/astrodino/data/dataset.py:79
↓ 5 callersMethod__init__
( self, in_features: int, hidden_features: int, activation: Optional[Callable]
astroclip/modules.py:79
↓ 4 callersFunctionis_enabled
Returns: True if distributed training is enabled
astroclip/astrodino/distributed.py:19
↓ 3 callersMethod__init__
(self, n_in, n_out, n_hidden=(16, 16, 16), act=None, dropout=0)
downstream_tasks/property_estimation/baselines/modules.py:104
↓ 3 callersMethod_use_ddp_or_ddp2
(trainer: Trainer)
astroclip/models/moco_v2.py:282
↓ 3 callersMethodforward_without_preprocessing
Forward pass through the model. The training step performs masking before preprocessing, thus samples should not be preprocessed again
astroclip/models/specformer.py:56
↓ 3 callersMethodmax
(self)
astroclip/astrodino/utils.py:202
↓ 3 callersMethodpreprocess
(self, x)
astroclip/models/specformer.py:119
↓ 2 callersMethod__init__
Cross-attention image module that takes token outputs from the AstroDINO model and passes them through a cross-attention mechanism an
astroclip/models/astroclip.py:158
↓ 2 callersMethod__init__
(self, scales=None, m=0.03, Q=20, bands=["g", "r", "z"])
astroclip/astrodino/data/augmentations.py:141
↓ 2 callersFunction_check_env_variable
(key: str, new_value: str)
astroclip/astrodino/distributed.py:149
↓ 2 callersMethod_dequeue_and_enqueue
(self, keys, queue_ptr, queue)
astroclip/models/moco_v2.py:91
↓ 2 callersFunction_init_by_depth
Initialize the weights of a module based on the depth of the model.
astroclip/modules.py:341
↓ 2 callersFunctiondo_test
(cfg, model, iteration)
astroclip/astrodino/trainer.py:136
↓ 2 callersMethodmask_sequence
Mask batched sequence
astroclip/models/specformer.py:115
↓ 2 callersMethodsetup
(self, stage: str)
astroclip/data/datamodule.py:25
↓ 2 callersFunctionsetup_astrodino
Set up AstroDINO model
astroclip/astrodino/utils.py:219
↓ 2 callersMethodtrain_dataloader
(self)
astroclip/data/datamodule.py:29
↓ 2 callersMethodupdate
(self, value, num=1)
astroclip/astrodino/utils.py:168
↓ 2 callersMethodval_dataloader
(self)
astroclip/data/datamodule.py:39
↓ 1 callersMethod__init__
( self, center_crop: int = 144, bands: List[str] = ["g", "r", "z"], m: float =
astroclip/data/datamodule.py:50
↓ 1 callersMethod__init__
( self, *, split: "LegacySurvey.Split", root: str, extra: str = None,
astroclip/astrodino/data/dataset.py:36
↓ 1 callersMethod__init__
(self, n_in, n_out, n_hidden=(16, 16, 16), act=None, dropout=0)
downstream_tasks/property_estimation/property_utils/models.py:78
↓ 1 callersMethod_batch_shuffle_ddp
Batch shuffle, for making use of BatchNorm. *** Only support DistributedDataParallel (DDP) model. ***
astroclip/models/moco_v2.py:108
↓ 1 callersMethod_batch_unshuffle_ddp
Undo batch shuffle. *** Only support DistributedDataParallel (DDP) model. ***
astroclip/models/moco_v2.py:136
↓ 1 callersFunction_collect_env_vars
()
astroclip/astrodino/distributed.py:120
↓ 1 callersMethod_conv_blocks
(self, filters, sizes, dropout=0)
downstream_tasks/property_estimation/baselines/modules.py:157
↓ 1 callersFunction_cross_match_tables
Cross-match two tables.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:49
↓ 1 callersFunction_download_data
Download the PROVABGS data from the web and save it to the specified directory.
downstream_tasks/property_estimation/property_utils/cross_match.py:21
↓ 1 callersFunction_download_gz5_decals
Download Galaxy Zoo 5 classifications.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:93
↓ 1 callersMethod_downsample
(self, x)
downstream_tasks/property_estimation/baselines/modules.py:176
↓ 1 callersFunction_generate_catalog
Generate a catalog from a list of files.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:27
↓ 1 callersFunction_get_available_port
()
astroclip/astrodino/distributed.py:101
↓ 1 callersFunction_get_best_fit
Get the best fit model for each galaxy.
downstream_tasks/property_estimation/property_utils/cross_match.py:37
↓ 1 callersFunction_get_data
(train_path, test_path, source)
downstream_tasks/property_estimation/posterior_estimation.py:51
↓ 1 callersFunction_get_file_location
Get the locations of the Legacy Survey image files.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:100
↓ 1 callersFunction_get_images
Get images from files.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:71
↓ 1 callersFunction_get_master_port
(seed: int = 0)
astroclip/astrodino/distributed.py:90
↓ 1 callersFunction_get_predictions
Use model to get predictions
downstream_tasks/property_estimation/baselines/trainer.py:19
↓ 1 callersFunction_get_properties
(train_dataset, test_dataset, properties=PROPERTIES)
downstream_tasks/property_estimation/posterior_estimation.py:34
↓ 1 callersMethod_initialize_model
(self, model_name)
downstream_tasks/property_estimation/baselines/modules.py:46
↓ 1 callersFunction_is_slurm_job_process
()
astroclip/astrodino/distributed.py:128
↓ 1 callersFunction_make_sampler
( *, dataset, type: Optional[SamplerType] = None, shuffle: bool = False, seed: int = 0,
astroclip/astrodino/data/loaders.py:82
↓ 1 callersMethod_mask_seq
Randomly masks contiguous sections of an unbatched sequence, ensuring separation between chunks is at least chunk_width.
astroclip/models/specformer.py:155
↓ 1 callersMethod_momentum_update_key_encoder
Momentum update of the key encoder.
astroclip/models/moco_v2.py:82
↓ 1 callersFunction_parse_dataset_str
(dataset_str: str)
astroclip/astrodino/data/loaders.py:24
↓ 1 callersFunction_parse_slurm_node_list
(s: str)
astroclip/astrodino/distributed.py:132
↓ 1 callersMethod_process_images
(self, images)
astroclip/data/datamodule.py:60
↓ 1 callersMethod_reset_parameters_datapt
(self)
astroclip/models/specformer.py:128
↓ 1 callersFunction_restrict_print_to_main_process
This function disables printing when not in the main process
astroclip/astrodino/distributed.py:74
↓ 1 callersMethod_set_from_local
(self)
astroclip/astrodino/distributed.py:216
↓ 1 callersMethod_set_from_preset_env
(self)
astroclip/astrodino/distributed.py:204
↓ 1 callersMethod_set_from_slurm_env
(self)
astroclip/astrodino/distributed.py:187
↓ 1 callersMethod_slice
(self, x)
astroclip/models/specformer.py:138
↓ 1 callersFunctionapply_optim_scheduler
(optimizer, lr, wd, last_layer_lr)
astroclip/astrodino/trainer.py:127
↓ 1 callersFunctionbuild_optimizer
(cfg, params_groups)
astroclip/astrodino/trainer.py:73
↓ 1 callersFunctionbuild_schedulers
(cfg)
astroclip/astrodino/trainer.py:79
↓ 1 callersFunctioncross_match_galaxy_zoo
Pairs Galaxy Zoo classifications with DECaLS images in an Astropy table. Args: root_dir (str): Root directory of DECaLS images.
downstream_tasks/morphology_classification/morphology_utils/cross_match.py:124
↓ 1 callersFunctioncross_match_provabgs
Cross-match the AstroCLIP and PROVABGS datasets.
downstream_tasks/property_estimation/property_utils/cross_match.py:74
↓ 1 callersFunctiondecals_to_rgb
(image, bands=["g", "r", "z"], scales=None, m=0.03, Q=20.0)
astroclip/astrodino/embed_legacysurvey/embed_legacysurvey.py:24
↓ 1 callersFunctiondefault_dotenv_values
Use a default .env but tell the user how to create their own.
astroclip/env.py:17
↓ 1 callersFunctiondo_train
(cfg, model, run_name, group_name, resume=False)
astroclip/astrodino/trainer.py:154
↓ 1 callersMethoddump_in_output_file
(self, iteration, iter_time, data_time)
astroclip/astrodino/utils.py:59
↓ 1 callersFunctionembed_astroclip
Extract embeddings from the AstroClip model and save them to a file
downstream_tasks/similarity_search/embed_astroclip.py:13
↓ 1 callersFunctionembed_galaxy_zoo
( galaxy_zoo_file: str, pretrained_dir: str, batch_size: int = 128, )
downstream_tasks/morphology_classification/embed_galaxy_zoo.py:55
↓ 1 callersFunctionembed_legacysurvey
( dset_root: str, save_dir: str, astrodino_dir: str, batch_size=512, num_gpus=4 )
astroclip/astrodino/embed_legacysurvey/embed_legacysurvey.py:96
↓ 1 callersFunctionembed_provabgs
( provabgs_file_train: str, provabgs_file_test: str, pretrained_dir: str, batch_size: int = 51
downstream_tasks/property_estimation/embed_provabgs.py:66
↓ 1 callersMethodexport
(self, *, overwrite: bool)
astroclip/astrodino/distributed.py:225
↓ 1 callersFunctionget_args_parser
(add_help: bool = True)
astroclip/astrodino/trainer.py:37
↓ 1 callersFunctionget_embeddings
Get embeddings for images using models
downstream_tasks/morphology_classification/embed_galaxy_zoo.py:20
↓ 1 callersFunctionget_embeddings
Get embeddings for images using models
downstream_tasks/property_estimation/embed_provabgs.py:17
↓ 1 callersFunctionget_global_rank
Returns: The rank of the current process within the global process group.
astroclip/astrodino/distributed.py:35
↓ 1 callersMethodget_logits
( self, image_features: torch.FloatTensor, spectrum_features: torch.FloatTensor,
astroclip/models/astroclip.py:118
↓ 1 callersFunctionimport_package
Import a torch package from a given path
astroclip/astrodino/embed_legacysurvey/embed_legacysurvey.py:37
↓ 1 callersMethodinit_encoders
Override to add your own encoders.
astroclip/models/moco_v2.py:72
↓ 1 callersFunctionis_main_process
Returns: True if the current process is the main one.
astroclip/astrodino/distributed.py:66
↓ 1 callersMethodlog_every
( self, iterable, print_freq, header=None, n_iterations=None, start_iteration=0 )
astroclip/astrodino/utils.py:80
↓ 1 callersFunctionmain_cli
(args: ArgsType = None, run: bool = True)
astroclip/trainer.py:44
↓ 1 callersFunctionmain_cli
(cli_args=None)
astroclip/astrodino/trainer.py:346
↓ 1 callersFunctionmake_data_loader
Creates a data loader with the specified parameters. Args: dataset: A dataset (third party, LaViDa or WebDataset). batch_siz
astroclip/astrodino/data/loaders.py:148
↓ 1 callersFunctionmake_dataset
Creates a dataset with the specified parameters. Args: dataset_str: A dataset string description (e.g. ImageNet:split=TRAIN).
astroclip/astrodino/data/loaders.py:49
↓ 1 callersMethodplot_spectrum
(self, batch, output)
astroclip/callbacks.py:62
↓ 1 callersMethodreset_parameters
(self)
astroclip/modules.py:285
↓ 1 callersFunctionrun_flow_for_model
( source: str, train_path: str, test_path: str, num_ndes: int = 10, max_hidden: int = 32,
downstream_tasks/property_estimation/posterior_estimation.py:231
↓ 1 callersMethodsetup
(self, stage=None)
downstream_tasks/property_estimation/baselines/data.py:33
↓ 1 callersMethodsynchronize_between_processes
(self)
astroclip/astrodino/utils.py:52
↓ 1 callersMethodtest_dataloader
(self)
downstream_tasks/property_estimation/baselines/data.py:100
↓ 1 callersFunctiontrain_baseline
( train_dataset: str, test_dataset: str, save_path: str, modality: str, model_name: str,
downstream_tasks/property_estimation/baselines/trainer.py:39
↓ 1 callersFunctiontrain_flow
( train_loader, val_loader, test_loader, base_dist, transform, index, num_epochs=1
downstream_tasks/property_estimation/posterior_estimation.py:142
Method__call__
(self, samples)
astroclip/data/datamodule.py:71
Method__call__
(self, image)
astroclip/astrodino/data/augmentations.py:86
Method__call__
(self, imgs)
astroclip/astrodino/data/augmentations.py:161
Method__call__
(self, image: np.ndarray)
astroclip/astrodino/data/augmentations.py:220
Method__call__
(self, image: np.ndarray)
astroclip/astrodino/data/augmentations.py:279
Method__getattr__
(self, attr)
astroclip/astrodino/utils.py:37
Method__getitem__
(self, index: int)
astroclip/astrodino/data/dataset.py:82
Method__getitem__
(self, index: int)
astroclip/astrodino/data/dataset.py:156
next →1–100 of 197, ranked by callers