Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WillDreamer/Aurora
/ functions
Functions
226 in github.com/WillDreamer/Aurora
⨍
Functions
226
◇
Types & classes
41
Function
level_to_args
(level)
transform/randaugment.py:203
Method
load_decompress_img_from_lmdb_value
Args: lmdb_value: image binary from with open(filepath, "rb") as f: lmdb_valu
data/vqa_dataset.py:94
Method
load_pretrained
(self, checkpoint_path, prefix='')
CP/vit.py:247
Method
no_weight_decay
(self)
CP/vit.py:224
Method
no_weight_decay
(self)
CP/med.py:493
Function
none_level_to_args
(level)
transform/randaugment.py:241
Function
posterize_func
same output as PIL.ImageOps.posterize
transform/randaugment.py:173
Method
prepare_inputs_for_generation
(self, input_ids, past=None, attention_mask=None, **model_kwargs)
CP/med.py:1060
Function
rotate_func
like PIL, rotate by degree, not radians
transform/randaugment.py:66
Method
save_attention_map
(self, attention_map)
CP/vit.py:86
Method
save_attn_gradients
(self, attn_gradients)
CP/vit.py:80
Method
save_attn_gradients
(self, attn_gradients)
CP/med.py:140
Function
save_on_master
(*args, **kwargs)
utils.py:251
Method
set_input_embeddings
(self, value)
CP/med.py:717
Method
set_output_embeddings
(self, new_embeddings)
CP/med.py:949
Function
sharpness_func
The differences the this result and PIL are all on the 4 boundaries, the center areas are same
transform/randaugment.py:125
Function
shear_x_func
(img, factor, fill=(0, 0, 0))
transform/randaugment.py:146
Function
shear_y_func
(img, factor, fill=(0, 0, 0))
transform/randaugment.py:181
Function
solarize_func
same output as PIL.ImageOps.posterize
transform/randaugment.py:77
Function
step_lr_schedule
Decay the learning rate
utils.py:15
Method
synchronize_between_processes
Warning: does not synchronize the deque!
utils.py:49
Function
translate_x_func
same output as PIL.Image.transform
transform/randaugment.py:153
Function
translate_y_func
same output as PIL.Image.transform
transform/randaugment.py:163
Function
vqa_collate_fn
(batch)
data/videovqa_dataset.py:106
Function
vqa_collate_fn
(batch)
data/vqa_dataset.py:109
Function
warmup_lr_schedule
Warmup the learning rate
utils.py:9
← previous
201–226 of 226, ranked by callers