Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HazyResearch/butterfly
/ functions
Functions
1,242 in github.com/HazyResearch/butterfly
⨍
Functions
1,242
◇
Types & classes
260
Method
__iter__
(self)
cnn/imagenet/mixup.py:29
Method
__repr__
(self)
convolution/tee.py:100
Function
_alr
(optimizer, iteration, epoch)
cnn/imagenet/training.py:128
Function
_lr_fn
(iteration, epoch)
cnn/imagenet/training.py:140
Function
_make_scheduler
(args)
learning_transforms/tune.py:35
Method
_restore
(self, checkpoint_path)
learning_transforms/training.py:27
Method
_restore
(self, checkpoint_path)
cnn/distill_experiment.py:141
Method
_restore
(self, checkpoint_path)
cnn/distill_cov_experiment.py:169
Method
_restore
(self, checkpoint_path)
cnn/cifar_experiment.py:141
Method
_restore
(self, checkpoint_path)
cnn/permuted_experiment.py:235
Method
_restore
(self, checkpoint_path)
transformer/dynamic_conv_experiment.py:150
Method
_save
(self, checkpoint_dir)
learning_transforms/training.py:20
Method
_save
(self, checkpoint_dir)
cnn/distill_experiment.py:133
Method
_save
(self, checkpoint_dir)
cnn/distill_cov_experiment.py:161
Method
_save
(self, checkpoint_dir)
cnn/cifar_experiment.py:129
Method
_save
(self, checkpoint_dir)
cnn/permuted_experiment.py:218
Method
_save
(self, checkpoint_dir)
transformer/dynamic_conv_experiment.py:146
Function
_sbc
(epoch)
cnn/imagenet/utils.py:9
Method
_setup
(self, config)
learning_transforms/learning_hadamard.py:56
Method
_setup
(self, config)
learning_transforms/learning_legendre.py:35
Method
_setup
(self, config)
learning_transforms/learning_legendre.py:82
Method
_setup
(self, config)
learning_transforms/learning_vandermonde.py:34
Method
_setup
(self, config)
learning_transforms/learning_vandermonde.py:79
Method
_setup
(self, config)
learning_transforms/learning_ops.py:33
Method
_setup
(self, config)
learning_transforms/learning_fft.py:40
Method
_setup
(self, config)
learning_transforms/learning_fft.py:61
Method
_setup
(self, config)
learning_transforms/learning_fft.py:85
Method
_setup
(self, config)
learning_transforms/learning_fft.py:121
Method
_setup
(self, config)
learning_transforms/learning_fft.py:397
Method
_setup
(self, config)
learning_transforms/learning_fft.py:442
Method
_setup
(self, config)
learning_transforms/learning_fft.py:457
Method
_setup
(self, config)
learning_transforms/learning_fft.py:480
Method
_setup
(self, config)
learning_transforms/learning_fft.py:523
Method
_setup
(self, config)
learning_transforms/learning_circulant.py:35
Method
_setup
(self, config)
learning_transforms/learning_circulant.py:80
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:64
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:85
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:109
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:145
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:421
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:466
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:488
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:513
Method
_setup
(self, config)
learning_transforms/old/learning_fft_old.py:553
Method
_setup
(self, config)
cnn/distill_experiment.py:40
Method
_setup
(self, config)
cnn/distill_cov_experiment.py:41
Method
_setup
(self, config)
cnn/cifar_experiment.py:36
Method
_setup
(self, config)
cnn/permuted_experiment.py:37
Method
_setup
(self, config)
transformer/dynamic_conv_experiment.py:51
Function
_step
(input, target, optimizer_step=True, sync_metrics=False)
cnn/imagenet/training.py:199
Function
_timed_function
(*args, **kwargs)
cnn/imagenet/utils.py:35
Method
_train
(self)
learning_transforms/learning_hadamard.py:67
Method
_train
(self)
learning_transforms/learning_legendre.py:67
Method
_train
(self)
learning_transforms/learning_legendre.py:110
Method
_train
(self)
learning_transforms/learning_vandermonde.py:64
Method
_train
(self)
learning_transforms/learning_vandermonde.py:108
Method
_train
(self)
learning_transforms/learning_ops.py:48
Method
_train
(self)
learning_transforms/learning_fft.py:49
Method
_train
(self)
learning_transforms/learning_fft.py:71
Method
_train
(self)
learning_transforms/learning_fft.py:97
Method
_train
(self)
learning_transforms/learning_fft.py:109
Method
_train
(self)
learning_transforms/learning_fft.py:129
Method
_train
(self)
learning_transforms/learning_fft.py:141
Method
_train
(self)
learning_transforms/learning_fft.py:427
Method
_train
(self)
learning_transforms/learning_fft.py:507
Method
_train
(self)
learning_transforms/learning_fft.py:537
Method
_train
(self)
learning_transforms/learning_circulant.py:65
Method
_train
(self)
learning_transforms/learning_circulant.py:109
Method
_train
(self)
learning_transforms/training.py:44
Method
_train
(self)
learning_transforms/training.py:100
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:73
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:95
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:121
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:133
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:153
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:165
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:451
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:476
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:501
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:525
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:537
Method
_train
(self)
learning_transforms/old/learning_fft_old.py:567
Method
_train
(self)
cnn/distill_experiment.py:129
Method
_train
(self)
cnn/distill_cov_experiment.py:121
Method
_train
(self)
cnn/cifar_experiment.py:113
Method
_train
(self)
cnn/permuted_experiment.py:209
Method
_train
(self)
transformer/dynamic_conv_experiment.py:108
Function
_weights_init
(m)
convolution/models/resnet_cifar.py:16
Function
_weights_init
(m)
cnn/models/resnet_original.py:39
Function
_worker_init_fn
(id)
cnn/imagenet_main.py:190
Function
_worker_init_fn
(id)
cnn/imagenet_finetune.py:183
Function
accuracy
Computes the precision@k for the specified values of k
cnn/imagenet/utils.py:42
Function
acdc
Construct an nn.Module based on Butterfly that exactly performs either the multiplication: x -> diag2 @ iDCT @ diag1 @ DCT @ x or
torch_butterfly/special.py:714
Function
adam
()
cnn/cifar_experiment.py:191
Function
add_gumbel_noise
Args: log_alpha: shape (N, N) temp: temperature parameter, a float. sample_shape: represents shape of independent draws Returns:
cnn/permutation_utils.py:275
Function
add_prefix
(state_dict, prefix)
cnn/train_utils.py:13
Method
allreduce_hook
(*unused)
cnn/distributed.py:71
Method
allreduce_params
()
cnn/distributed.py:62
Method
argmax
Return: p: (self.size, ) array of int, the most probable permutation.
butterfly/permutation.py:103
Method
argmax
Return: p: (self.size, ) array of int, the most probable permutation.
learning_transforms/butterfly_old.py:630
← previous
next →
601–700 of 1,242, ranked by callers