MCPcopy Create free account

hub / github.com/PolymathicAI/AstroCLIP / functions

Functions197 in github.com/PolymathicAI/AstroCLIP

⨍ Functions 197◇ Types & classes 44
Method__init__
( self, optimizer: torch.optim.Optimizer, T_max: int, T_warmup: int = 0,
astroclip/scheduler.py:23
Method__init__
( self, embed_dim: int, n_head: int, model_embed_dim: int, dropout: fl
astroclip/modules.py:28
Method__init__
( self, embedding_dim: int, num_heads: int, causal: bool, dropout: flo
astroclip/modules.py:125
Method__init__
( self, embedding_dim: int, num_heads: int, causal: bool, dropout: flo
astroclip/modules.py:223
Method__init__
( self, shape: Union[int, Tuple[int, ...], torch.Size], eps: float = 1e-5, bia
astroclip/modules.py:265
Method__init__
( self, weight: Union[torch.Tensor, nn.Parameter], bias: Union[None, torch.Tensor, nn.
astroclip/modules.py:314
Method__init__
(self)
astroclip/callbacks.py:59
Method__init__
( self, path: str, columns: List[str] = ["image", "spectrum"], batch_size: int
astroclip/data/datamodule.py:14
Method__init__
( self, base_encoder: Union[str, torch.nn.Module] = "resnet18", emb_dim: int = 128,
astroclip/models/moco_v2.py:21
Method__init__
The AstroCLIP model that takes an image and a spectrum and embeds them into a common space using CLIP loss. Note that you must provid
astroclip/models/astroclip.py:17
Method__init__
Cross-attention spectrum module that takes a spectrum and passes it through a pretrained SpecFormer model and then through a cross-at
astroclip/models/astroclip.py:239
Method__init__
( self, input_dim: int, embed_dim: int, num_layers: int, num_heads: in
astroclip/models/specformer.py:14
Method__init__
(self, delimiter="\t", wandb=None, output_file=None)
astroclip/astrodino/utils.py:24
Method__init__
(self, window_size=20, fmt=None)
astroclip/astrodino/utils.py:160
Method__init__
(self)
astroclip/astrodino/distributed.py:158
Method__init__
( self, *, split: "LegacySurvey.Split", root: str, extra: str = None,
astroclip/astrodino/data/dataset.py:118
Method__init__
( self, global_crops_scale, local_crops_scale, local_crops_number, glo
astroclip/astrodino/data/augmentations.py:19
Method__init__
(self, *, p: float = 0.5)
astroclip/astrodino/data/augmentations.py:117
Method__init__
(self, *, im_dim=144, p: float = 0.5)
astroclip/astrodino/data/augmentations.py:126
Method__init__
( self, scaling: List = [1.0], mean: float = 0, im_dim: int = 144, im_
astroclip/astrodino/data/augmentations.py:194
Method__init__
( self, scaling: List = [1.0], im_dim: int = 144, im_ch: int = 3, deca
astroclip/astrodino/data/augmentations.py:256
Method__init__
( self, input_dim: int, num_classes: int, hidden_dim: int, dropout_rate: int )
downstream_tasks/morphology_classification/morphology_utils/models.py:14
Method__init__
( self, model_name, modality, properties, scale, num_epochs,
downstream_tasks/property_estimation/baselines/modules.py:18
Method__init__
(self, n_out=1)
downstream_tasks/property_estimation/baselines/modules.py:89
Method__init__
(self, n_latent, n_hidden=(32, 32), act=None, dropout=0)
downstream_tasks/property_estimation/baselines/modules.py:131
Method__init__
( self, train_data: Table, test_data: Table, modality: str, properties
downstream_tasks/property_estimation/baselines/data.py:9
Method__init__
( self, input_dim: int, context_dim: int, hidden_dims: int, num_flows:
downstream_tasks/property_estimation/property_utils/models.py:98
Method__len__
(self)
astroclip/astrodino/data/dataset.py:95
Method__len__
(self)
astroclip/astrodino/data/dataset.py:169
Method__str__
(self)
astroclip/astrodino/utils.py:46
Method__str__
(self)
astroclip/astrodino/utils.py:209
Method_attention_hook
(self, grad)
downstream_tasks/property_estimation/baselines/modules.py:211
Method_generate_examples
Yields examples.
astroclip/data/dataset.py:95
Method_info
(self)
astroclip/data/dataset.py:54
Function_safe_eval
Safely evaluate an arithmetic expression. :param s: expression to evaluate; should only contain numbers, spaces, or the symbols `+, -, *,
astroclip/callbacks.py:11
Method_split_generators
(self, dl_manager)
astroclip/data/dataset.py:77
Methodadd_meter
(self, name, meter)
astroclip/astrodino/utils.py:56
Methodadd_model_specific_args
(parent_parser)
astroclip/models/moco_v2.py:256
Methodattention_grad
(self)
downstream_tasks/property_estimation/baselines/modules.py:215
Methodavg
(self)
astroclip/astrodino/utils.py:193
Methodbefore_instantiate_classes
(self)
astroclip/trainer.py:21
Methodconfigure_optimizers
( lightning_module: LightningModule, optimizer: Optimizer, lr_scheduler: Optional[LRSc
astroclip/trainer.py:26
Methodconfigure_optimizers
(self)
astroclip/models/moco_v2.py:242
Methodconfigure_optimizers
(self)
downstream_tasks/property_estimation/baselines/modules.py:80
Functionenable
Enable distributed mode Args: set_cuda_current_device: If True, call torch.cuda.set_device() to set the current PyTorch CUDA
astroclip/astrodino/distributed.py:245
Methodextra_repr
(self)
astroclip/modules.py:333
Functionfew_shot
Train a few-shot model using a simple neural network
downstream_tasks/property_estimation/property_utils/models.py:15
Methodforward
(self, x: torch.tensor)
astroclip/modules.py:47
Methodforward
(self, x: torch.Tensor)
astroclip/modules.py:99
Methodforward
(self, x: torch.Tensor)
astroclip/modules.py:160
Methodforward
(self, x)
astroclip/modules.py:243
Methodforward
(self, input)
astroclip/modules.py:290
Methodforward
(self, x: torch.Tensor)
astroclip/modules.py:330
Methodforward
Input: im_q: a batch of query images im_k: a batch of key images queue: a queue from which to pick negati
astroclip/models/moco_v2.py:154
Methodforward
( self, input: torch.Tensor, input_type: str, )
astroclip/models/astroclip.py:60
Methodforward
( self, image_features: torch.FloatTensor, spectrum_features: torch.FloatTensor,
astroclip/models/astroclip.py:134
Methodforward
(self, x: torch.tensor, return_weights: bool = False)
astroclip/models/astroclip.py:218
Methodforward
( self, x: torch.tensor, y: torch.tensor = None, return_weights: bool = False )
astroclip/models/astroclip.py:289
Methodforward
Forward pass through the model.
astroclip/models/specformer.py:51
Methodforward
(self, x: torch.tensor)
downstream_tasks/morphology_classification/morphology_utils/models.py:34
Methodforward
(self, x)
downstream_tasks/property_estimation/baselines/modules.py:61
Methodforward
(self, x)
downstream_tasks/property_estimation/baselines/modules.py:97
Methodforward
(self, y)
downstream_tasks/property_estimation/baselines/modules.py:189
Functionget_global_size
Returns: The number of processes in the process group
astroclip/astrodino/distributed.py:27
Functionget_local_rank
Returns: The rank of the current process within the local (per-machine) process group.
astroclip/astrodino/distributed.py:43
Functionget_local_size
Returns: The size of the per-machine process group, i.e. the number of processes per machine.
astroclip/astrodino/distributed.py:54
Methodget_lr
(self)
astroclip/scheduler.py:38
Methodglobal_avg
(self)
astroclip/astrodino/utils.py:198
Methodlength
(self)
astroclip/astrodino/data/dataset.py:23
Methodlength
(self)
astroclip/astrodino/data/dataset.py:105
Functionload_model
(repo_id, filename)
astroclip/models/loader.py:5
Methodlog_hyperparams
(self, *args, **kwargs)
astroclip/callbacks.py:39
Methodmedian
(self)
astroclip/astrodino/utils.py:188
Methodn_parameters
(self)
downstream_tasks/property_estimation/baselines/modules.py:208
Methodon_validation_batch_start
( self, trainer: Trainer, pl_module: LightningModule, batch: Any, batc
astroclip/callbacks.py:86
Functionplot_radar
Functionality for plotting radar chart
downstream_tasks/morphology_classification/morphology_utils/plotting.py:8
Functionplot_scatter
Functionality to plot redshift scatter plots for different models.
downstream_tasks/property_estimation/property_utils/plotting.py:7
Functionplot_similar_images
Functionality for plotting retrieved galaxy images
downstream_tasks/similarity_search/plotting.py:8
Functionplot_similar_spectra
Functionality for plotting retrieved galaxy spectra
downstream_tasks/similarity_search/plotting.py:34
Functionprocess_file
Process a single file in the dataset
astroclip/astrodino/embed_legacysurvey/embed_legacysurvey.py:44
Methodsave_config
( self, trainer: Trainer, pl_module: LightningModule, stage: str )
astroclip/callbacks.py:48
Methodsplit
(self)
astroclip/astrodino/data/dataset.py:153
Methodsynchronize_between_processes
Distributed synchronization of the metric Warning: does not synchronize the deque!
astroclip/astrodino/utils.py:173
Methodtrain_dataloader
(self)
downstream_tasks/property_estimation/baselines/data.py:94
Functiontrain_eval_on_question
Function to train and evaluate a simple feedforward neural network on a dataset.
downstream_tasks/morphology_classification/morphology_utils/models.py:40
Methodtraining_step
(self, batch, batch_idx)
astroclip/models/moco_v2.py:200
Methodtraining_step
(self, batch, batch_idx)
astroclip/models/astroclip.py:74
Methodtraining_step
(self, batch)
astroclip/models/specformer.py:82
Methodtraining_step
(self, batch, batch_idx)
downstream_tasks/property_estimation/baselines/modules.py:64
Methodval_dataloader
(self)
downstream_tasks/property_estimation/baselines/data.py:97
Methodvalidation_epoch_end
(self, outputs)
astroclip/models/moco_v2.py:238
Methodvalidation_step
(self, batch, batch_idx)
astroclip/models/moco_v2.py:220
Methodvalidation_step
(self, batch, batch_idx)
astroclip/models/astroclip.py:97
Methodvalidation_step
(self, batch)
astroclip/models/specformer.py:98
Methodvalidation_step
(self, batch, batch_idx)
downstream_tasks/property_estimation/baselines/modules.py:73
Methodvalue
(self)
astroclip/astrodino/utils.py:206
Functionzero_shot
Train a zero-shot model using KNN
downstream_tasks/property_estimation/property_utils/models.py:65
← previous101–197 of 197, ranked by callers