Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/RuslanKhalitov/ChordMixer
/ functions
Functions
207 in github.com/RuslanKhalitov/ChordMixer
⨍
Functions
207
◇
Types & classes
32
↓ 10 callers
Method
__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
models/chordmixer.py:32
↓ 3 callers
Method
__init__
(self, backbone, dim, linear_decoder=False, n_class=2)
trainer.py:216
↓ 3 callers
Method
_calculate_loss
(self, batch, mode)
trainer.py:66
↓ 3 callers
Method
_calculate_loss
(self, batch, mode)
trainer.py:156
↓ 3 callers
Method
_calculate_loss
(self, batch, mode)
trainer.py:266
↓ 3 callers
Method
get_bucket_boundaries
(self, df)
dataloaders/genbank.py:133
↓ 3 callers
Method
get_bucket_boundaries
(self, df)
dataloaders/lra_listops_var.py:149
↓ 3 callers
Method
get_bucket_boundaries
(self, df)
dataloaders/longdoc.py:129
↓ 3 callers
Method
get_bucket_boundaries
(self, df)
dataloaders/adding.py:87
↓ 3 callers
Function
map_scheduler
( scheduler_name, optimizer, num_warmup_steps, num_training_steps )
optim/register.py:3
↓ 3 callers
Function
polynomial_warmup_polynomial_decay
Adapted from https://github.com/huggingface/transformers/blob/v4.26.1/src/transformers/optimization.py#L173
optim/schedulers.py:122
↓ 3 callers
Method
train_dataloader
(self)
dataloaders/lra_image.py:84
↓ 2 callers
Function
linear_warmup_polynomial_decay
https://github.com/huggingface/transformers/blob/v4.26.1/src/transformers/optimization.py#L173
optim/schedulers.py:92
↓ 2 callers
Function
map_norm
Maps the given normalization type to the corresponding PyTorch module. Args: norm_type (str): The normalization type ('LN', 'BN', 'G
models/chordmixer.py:87
↓ 2 callers
Function
min_max_scale
(min_val, max_val, value)
optim/schedulers.py:4
↓ 2 callers
Method
process_dataset
(self)
dataloaders/genbank.py:60
↓ 2 callers
Method
process_dataset
(self)
dataloaders/lra_listops_var.py:61
↓ 2 callers
Method
process_dataset
(self)
dataloaders/lra_text.py:32
↓ 2 callers
Method
process_dataset
(self)
dataloaders/lra_retrieval.py:78
↓ 2 callers
Method
process_dataset
(self)
dataloaders/longdoc.py:55
↓ 2 callers
Method
process_dataset
(self)
dataloaders/lra_listops.py:69
↓ 1 callers
Function
PrepareGenbank
(data_path, taskname)
dataloaders/utils/preprocess_genbank.py:32
↓ 1 callers
Function
PrepareLongdoc
Adapted from https://github.com/wiedersehne/Paramixer/blob/main/long-document-clf/articles_preprocessing.py
dataloaders/utils/preprocess_longdoc.py:6
↓ 1 callers
Method
__init__
(self, df)
dataloaders/adding.py:20
↓ 1 callers
Method
_create_data_buckets
(self)
dataloaders/utils/bucket_sampler.py:36
↓ 1 callers
Method
_create_iter_list
(self, data_buckets)
dataloaders/utils/bucket_sampler.py:45
↓ 1 callers
Method
_get_bucket_id
(self, seq_length)
dataloaders/utils/bucket_sampler.py:75
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/genbank.py:115
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/lra_listops_var.py:127
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/lra_text.py:89
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/lra_retrieval.py:155
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/longdoc.py:111
↓ 1 callers
Method
_load_from_cache
(self)
dataloaders/lra_listops.py:135
↓ 1 callers
Method
_save_to_cache
(self, dataset)
dataloaders/genbank.py:108
↓ 1 callers
Method
_save_to_cache
(self, dataset, tokenizer, vocab)
dataloaders/lra_listops_var.py:116
↓ 1 callers
Method
_save_to_cache
(self, dataset, tokenizer, vocab)
dataloaders/lra_text.py:78
↓ 1 callers
Method
_save_to_cache
(self, dataset, tokenizer, vocab)
dataloaders/lra_retrieval.py:144
↓ 1 callers
Method
_save_to_cache
(self, dataset)
dataloaders/longdoc.py:104
↓ 1 callers
Method
_save_to_cache
(self, dataset, tokenizer, vocab)
dataloaders/lra_listops.py:124
↓ 1 callers
Function
adding
(sequences, base_length, min_len, max_len)
data/generate_adding.py:23
↓ 1 callers
Function
cli_main
(cfg: OmegaConf)
trainer.py:327
↓ 1 callers
Function
generator
(base_length)
data/generate_adding.py:54
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/genbank.py:126
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/lra_listops_var.py:142
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/lra_text.py:104
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/lra_retrieval.py:170
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/longdoc.py:122
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/adding.py:80
↓ 1 callers
Method
get_cache_dir
(self)
dataloaders/lra_listops.py:150
↓ 1 callers
Function
linear_warmup_cosine_decay
https://github.com/huggingface/transformers/blob/v4.26.1/src/transformers/optimization.py#L104
optim/schedulers.py:7
↓ 1 callers
Function
linear_warmup_linear_decay
https://github.com/huggingface/transformers/blob/v4.26.1/src/transformers/optimization.py#L75
optim/schedulers.py:74
↓ 1 callers
Function
map_dataset
( data_dir, taskname, num_workers, batch_size, diff_lengths )
dataloaders/register.py:13
↓ 1 callers
Function
sampler
Function to sample length values
data/generate_adding.py:13
↓ 1 callers
Method
total_steps
(self)
trainer.py:105
↓ 1 callers
Method
total_steps
(self)
trainer.py:192
↓ 1 callers
Method
total_steps
(self)
trainer.py:302
Function
LoadGrayscale
(path)
dataloaders/lra_pathfinder.py:13
Function
LoadGrayscale
(path)
dataloaders/lra_pathfinderx.py:17
Method
__getitem__
(self, idx)
dataloaders/lra_pathfinder.py:45
Method
__getitem__
(self, idx)
dataloaders/lra_pathfinderx.py:49
Method
__getitem__
(self, index)
dataloaders/adding.py:23
Method
__getitem__
Returns: tuple (sample, target)
dataloaders/utils/masking_var_lengths.py:22
Method
__init__
(self, backbone, cfg, args)
trainer.py:41
Method
__init__
(self, backbone, cfg, args)
trainer.py:131
Method
__init__
(self, backbone, cfg, args)
trainer.py:236
Method
__init__
(self, threshold=0.04, dist_sync_on_step=False)
optim/custom_metrics.py:5
Method
__init__
( self, data_dir, taskname, num_workers, batch_size, **kwargs,
dataloaders/genbank.py:24
Method
__init__
( self, data_dir, batch_size, num_workers, max_length=2048, **
dataloaders/lra_listops_var.py:28
Method
__init__
( self, data_dir, num_workers, batch_size )
dataloaders/lra_image.py:11
Method
__init__
(self, data_dir, transform=None)
dataloaders/lra_pathfinder.py:21
Method
__init__
( self, data_dir, num_workers, batch_size, )
dataloaders/lra_pathfinder.py:54
Method
__init__
(self, data_dir, transform=None)
dataloaders/lra_pathfinderx.py:25
Method
__init__
( self, data_dir, num_workers, batch_size )
dataloaders/lra_pathfinderx.py:58
Method
__init__
( self, data_dir, num_workers, batch_size )
dataloaders/lra_text.py:15
Method
__init__
( self, data_dir, num_workers, batch_size, **kwargs, )
dataloaders/lra_retrieval.py:17
Method
__init__
( self, data_dir, taskname, num_workers, batch_size, **kwargs,
dataloaders/longdoc.py:24
Method
__init__
( self, data_dir, taskname, num_workers, batch_size, **kwargs,
dataloaders/adding.py:31
Method
__init__
( self, data_dir, batch_size, num_workers, max_length=2048, ap
dataloaders/lra_listops.py:26
Method
__init__
(self, df, mask_ratio=0.2, mask_value=-1)
dataloaders/utils/masking_var_lengths.py:17
Method
__init__
(self, lengths, bucket_boundaries, batch_size=4)
dataloaders/utils/bucket_sampler.py:19
Method
__init__
(self, track_size)
models/chordmixer_circular_pad.py:19
Method
__init__
(self, embedding_size)
models/chordmixer.py:57
Method
__init__
(self, embedding_size, n_groups)
models/chordmixer.py:76
Method
__init__
(self, track_size)
models/chordmixer.py:120
Method
__init__
(self, track_size)
models/chordmixer.py:155
Method
__init__
( self, embedding_size, track_size, hidden_size, mlp_dropout,
models/chordmixer.py:204
Method
__init__
( self, max_seq_len, track_size, hidden_size, mlp_dropout, lay
models/chordmixer.py:260
Method
__init__
(self, embedding_size, output_size)
models/chordmixer.py:315
Method
__init__
(self, embedding_size, output_size)
models/chordmixer.py:336
Method
__init__
(self, input_size, output_size, embedding_type, decoder, max_seq_len,
models/chordmixer.py:363
Method
__init__
(self, input_size, output_size, embedding_type, decoder, max_seq_len,
models/chordmixer.py:435
Method
__iter__
(self)
dataloaders/utils/bucket_sampler.py:25
Method
__len__
(self)
dataloaders/lra_pathfinder.py:42
Method
__len__
(self)
dataloaders/lra_pathfinderx.py:46
Method
__len__
(self)
dataloaders/adding.py:27
Method
__len__
(self)
dataloaders/utils/masking_var_lengths.py:49
Method
__len__
(self)
dataloaders/utils/bucket_sampler.py:33
Method
_cache_dir_name
(self)
dataloaders/genbank.py:123
Method
_cache_dir_name
(self)
dataloaders/lra_listops_var.py:139
Method
_cache_dir_name
(self)
dataloaders/lra_text.py:101
next →
1–100 of 207, ranked by callers