MCPcopy Create free account

hub / github.com/ZhangYuanhan-AI/Bamboo / functions

Functions84 in github.com/ZhangYuanhan-AI/Bamboo

↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
Bamboo-Benchmark/models/resnet.py:144
↓ 3 callersFunctionextract
(model, loader)
Bamboo-Benchmark/main.py:26
↓ 3 callersFunctionparameters_string
(module)
Bamboo-Benchmark/utils.py:24
↓ 2 callersMethod__init__
( self, positional_embedding_flag=True, transformer_width=512, context_lengt
Bamboo-Benchmark/models/text_transformer.py:79
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000, with_neck=False, pretrain_path=None, enable_fc=False)
Bamboo-Benchmark/models/resnet.py:98
↓ 2 callersMethod__repr__
(self)
Bamboo-Benchmark/dataset.py:309
↓ 2 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
Bamboo-Benchmark/models/simple_tokenizer.py:16
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
Bamboo-Benchmark/models/resnet.py:20
↓ 2 callersMethodencode
(self, text)
Bamboo-Benchmark/models/simple_tokenizer.py:121
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
Bamboo-Benchmark/models/simple_tokenizer.py:38
↓ 2 callersFunctionhas_file_allowed_extension
Checks if a file is an allowed extension. Args: filename (string): path to a file extensions (iterable of strings): extensions to
Bamboo-Benchmark/dataset.py:61
↓ 2 callersMethodread
(self, filename)
Bamboo-Benchmark/dataset.py:41
↓ 2 callersFunctiontest
(net, feats, labels)
Bamboo-Benchmark/main.py:42
↓ 2 callersMethodupdate
(self, val, num=1)
Bamboo-Benchmark/utils.py:76
↓ 1 callersMethod__init__
(self)
Bamboo-Benchmark/dataset.py:32
↓ 1 callersMethod__len__
(self)
Bamboo-Benchmark/dataset.py:206
↓ 1 callersMethod__len__
(self)
Bamboo-Benchmark/dataset.py:306
↓ 1 callersMethod_find_classes
Finds the class folders in a dataset. Args: dir (string): Root directory path. Returns: tuple: (cla
Bamboo-Benchmark/dataset.py:153
↓ 1 callersMethod_init_ceph
(self)
Bamboo-Benchmark/dataset.py:35
↓ 1 callersFunction_init_dist_slurm
Initialize slurm distributed training environment. If argument ``port`` is not specified, then the master port will be system environment vari
Bamboo-Benchmark/main.py:90
↓ 1 callersMethod_init_memcached
(self)
Bamboo-Benchmark/dataset.py:175
↓ 1 callersMethodattention
(self, x: torch.Tensor)
Bamboo-Benchmark/models/text_transformer.py:45
↓ 1 callersFunctionbasic_clean
(text)
Bamboo-Benchmark/models/simple_tokenizer.py:50
↓ 1 callersMethodbpe
(self, token)
Bamboo-Benchmark/models/simple_tokenizer.py:80
↓ 1 callersMethodbuild_attention_mask
(self)
Bamboo-Benchmark/models/text_transformer.py:136
↓ 1 callersMethodcheckpoint_fwd
checkpoint forward
Bamboo-Benchmark/models/text_transformer.py:63
↓ 1 callersFunctioncreate_logger
(name, log_file, rank, level=logging.INFO)
Bamboo-Benchmark/utils.py:93
↓ 1 callersMethoddecode
(self, tokens)
Bamboo-Benchmark/models/simple_tokenizer.py:129
↓ 1 callersFunctiondefault_bpe
()
Bamboo-Benchmark/models/simple_tokenizer.py:11
↓ 1 callersMethodencode_text
(self, texts)
Bamboo-Benchmark/models/text_transformer.py:165
↓ 1 callersMethodforward_features
(self, x)
Bamboo-Benchmark/models/timmvit.py:27
↓ 1 callersFunctionget_pil_version
()
Bamboo-Benchmark/utils.py:14
↓ 1 callersMethodinitialize_parameters
(self)
Bamboo-Benchmark/models/text_transformer.py:120
↓ 1 callersFunctionload_clip_state_text_model
(model, ckpt_path)
Bamboo-Benchmark/models/text_transformer.py:189
↓ 1 callersFunctionload_state_vision_model
(model, ckpt_path)
Bamboo-Benchmark/models/resnet.py:185
↓ 1 callersFunctionmake_dataset
(dir, class_to_idx, extensions)
Bamboo-Benchmark/dataset.py:87
↓ 1 callersFunctionpil_loader
(img_str)
Bamboo-Benchmark/dataset.py:53
↓ 1 callersFunctionrandom_seed
(seed_value)
Bamboo-Benchmark/utils.py:49
↓ 1 callersMethodreset
(self)
Bamboo-Benchmark/utils.py:67
↓ 1 callersFunctiontest_build
()
Bamboo-Benchmark/models/timmvit.py:53
↓ 1 callersMethodtokenize
(self, texts: Union[str, List[str]], context_length: int = 77)
Bamboo-Benchmark/models/text_transformer.py:148
↓ 1 callersFunctiontrain_feat_split
(seg_num, train_feats)
Bamboo-Benchmark/main.py:285
↓ 1 callersFunctiontrain_labels_split
(seg_num, train_feats)
Bamboo-Benchmark/main.py:303
↓ 1 callersFunctionwhitespace_clean
(text)
Bamboo-Benchmark/models/simple_tokenizer.py:56
MethodBNFunc
(*args, **kwargs)
Bamboo-Benchmark/models/resnet.py:103
Method__getitem__
Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.
Bamboo-Benchmark/dataset.py:182
Method__getitem__
Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.
Bamboo-Benchmark/dataset.py:275
Method__init__
(self, length=0)
Bamboo-Benchmark/utils.py:63
Method__init__
(self, root, extensions, transform=None, target_transform=None)
Bamboo-Benchmark/dataset.py:133
Method__init__
(self, root, transform=None, target_transform=None)
Bamboo-Benchmark/dataset.py:245
Method__init__
(self, root, meta, transform=None)
Bamboo-Benchmark/dataset.py:253
Method__init__
(self, bpe_path: str = default_bpe())
Bamboo-Benchmark/models/simple_tokenizer.py:63
Method__init__
(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None)
Bamboo-Benchmark/models/text_transformer.py:32
Method__init__
(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, checkpoint: bool = False)
Bamboo-Benchmark/models/text_transformer.py:56
Method__init__
(self, model_name)
Bamboo-Benchmark/models/efficientnet.py:11
Method__init__
(self, num_classes=119035, pretrain_path=None, enable_fc=False)
Bamboo-Benchmark/models/timmvit.py:23
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
Bamboo-Benchmark/models/resnet.py:28
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
Bamboo-Benchmark/models/resnet.py:60
Method__repr__
(self)
Bamboo-Benchmark/dataset.py:209
Function_get_clones
(module, N)
Bamboo-Benchmark/models/timmvit.py:18
Function_init_dist_pytorch
(backend, **kwargs)
Bamboo-Benchmark/main.py:83
Method_init_memcached
(self)
Bamboo-Benchmark/dataset.py:268
Functionclip_text
(clip_pretrain_path=None, **kwargs)
Bamboo-Benchmark/models/text_transformer.py:248
Functionclosure
()
Bamboo-Benchmark/main.py:329
Functioncollect_env_info
()
Bamboo-Benchmark/utils.py:18
Methoddtype
(self)
Bamboo-Benchmark/models/text_transformer.py:145
Functionefficientnet_B4
Constructs a ResNet-50 model.
Bamboo-Benchmark/models/efficientnet.py:30
Methodforward
(self, x: torch.Tensor)
Bamboo-Benchmark/models/text_transformer.py:18
Methodforward
(self, x: torch.Tensor)
Bamboo-Benchmark/models/text_transformer.py:27
Methodforward
(self, x: torch.Tensor)
Bamboo-Benchmark/models/text_transformer.py:49
Methodforward
(self, x: torch.Tensor)
Bamboo-Benchmark/models/text_transformer.py:72
Methodforward
(self, texts)
Bamboo-Benchmark/models/text_transformer.py:185
Methodforward
(image)
Bamboo-Benchmark/models/efficientnet.py:22
Methodforward
(self, x)
Bamboo-Benchmark/models/timmvit.py:42
Methodforward
(self, x)
Bamboo-Benchmark/models/resnet.py:38
Methodforward
(self, x)
Bamboo-Benchmark/models/resnet.py:73
Methodforward
(self, x)
Bamboo-Benchmark/models/resnet.py:161
Functionget_lr
(optimizer)
Bamboo-Benchmark/utils.py:106
Functionis_image_file
Checks if a file is an allowed image extension. Args: filename (string): path to a file Returns: bool: True if the filename
Bamboo-Benchmark/dataset.py:75
Functionresnet101
r"""ResNet-101 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained
Bamboo-Benchmark/models/resnet.py:247
Functionresnet50
Constructs a ResNet-50 model.
Bamboo-Benchmark/models/resnet.py:240
Functionresnetal50
Constructs a ResNet-50 model.
Bamboo-Benchmark/models/resnet.py:259
Functiontimvit
(**kwargs)
Bamboo-Benchmark/models/timmvit.py:47
Functiontrim_key
(key)
Bamboo-Benchmark/dataset.py:20