Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Amazing-J/structural-transformer
/ functions
Functions
986 in github.com/Amazing-J/structural-transformer
⨍
Functions
986
◇
Types & classes
128
↓ 3 callers
Method
step
Update the model parameters based on current gradients. Optionally, will employ gradient modification or update learning rate.
opennmt-feature/utils/optimizers.py:212
↓ 3 callers
Method
step
Update the model parameters based on current gradients. Optionally, will employ gradient modification or update learning rate.
opennmt-cnn/utils/optimizers.py:212
↓ 3 callers
Method
step
Update the model parameters based on current gradients. Optionally, will employ gradient modification or update learning rate.
opennmt-avg/utils/optimizers.py:212
↓ 3 callers
Method
step
Update the model parameters based on current gradients. Optionally, will employ gradient modification or update learning rate.
opennmt-baseline/utils/optimizers.py:212
↓ 3 callers
Method
step
Update the model parameters based on current gradients. Optionally, will employ gradient modification or update learning rate.
opennmt-self/utils/optimizers.py:212
↓ 3 callers
Function
store_vocab_to_file
(vocab, filename)
opennmt-feature/preprocess.py:221
↓ 3 callers
Method
train
The main training loops. by iterating over training data (i.e. `train_iter_fct`) and running validation (i.e. iterating over `valid_iter_
opennmt-feature/trainer.py:110
↓ 3 callers
Method
train
The main training loops. by iterating over training data (i.e. `train_iter_fct`) and running validation (i.e. iterating over `valid_iter_
opennmt-cnn/trainer.py:110
↓ 3 callers
Method
train
The main training loops. by iterating over training data (i.e. `train_iter_fct`) and running validation (i.e. iterating over `valid_iter_
opennmt-avg/trainer.py:110
↓ 3 callers
Method
train
The main training loops. by iterating over training data (i.e. `train_iter_fct`) and running validation (i.e. iterating over `valid_iter_
opennmt-baseline/trainer.py:110
↓ 3 callers
Method
train
The main training loops. by iterating over training data (i.e. `train_iter_fct`) and running validation (i.e. iterating over `valid_iter_
opennmt-self/trainer.py:110
↓ 2 callers
Method
__init__
(self, criterion, generator, normalization="sents")
opennmt-feature/utils/loss.py:220
↓ 2 callers
Method
__init__
(self, criterion, generator, normalization="sents")
opennmt-cnn/utils/loss.py:220
↓ 2 callers
Method
__init__
(self, head_count, model_dim, dropout=0.1)
opennmt-cnn/onmt/sublayer.py:26
↓ 2 callers
Method
__init__
(self, criterion, generator, normalization="sents")
opennmt-avg/utils/loss.py:220
↓ 2 callers
Method
__init__
(self, head_count, model_dim, dropout=0.1)
opennmt-avg/onmt/sublayer.py:26
↓ 2 callers
Method
__init__
(self, criterion, generator, normalization="sents")
opennmt-baseline/utils/loss.py:220
↓ 2 callers
Method
__init__
(self, criterion, generator, normalization="sents")
opennmt-self/utils/loss.py:220
↓ 2 callers
Method
__init__
(self, head_count, model_dim, dropout=0.1)
opennmt-self/onmt/sublayer.py:26
↓ 2 callers
Method
_compute_loss
Compute the loss. Subclass must define this method. Args: batch: the current batch. output: the predict out
opennmt-feature/utils/loss.py:90
↓ 2 callers
Method
_compute_loss
Compute the loss. Subclass must define this method. Args: batch: the current batch. output: the predict out
opennmt-cnn/utils/loss.py:90
↓ 2 callers
Method
_compute_loss
Compute the loss. Subclass must define this method. Args: batch: the current batch. output: the predict out
opennmt-avg/utils/loss.py:90
↓ 2 callers
Method
_compute_loss
Compute the loss. Subclass must define this method. Args: batch: the current batch. output: the predict out
opennmt-baseline/utils/loss.py:90
↓ 2 callers
Method
_compute_loss
Compute the loss. Subclass must define this method. Args: batch: the current batch. output: the predict out
opennmt-self/utils/loss.py:90
↓ 2 callers
Method
_compute_topk_scores_and_seq
(self, sequences, scores, scores_to_gather, flags)
opennmt-feature/onmt/beam.py:58
↓ 2 callers
Method
_compute_topk_scores_and_seq
(self, sequences, scores, scores_to_gather, flags)
opennmt-cnn/onmt/beam.py:58
↓ 2 callers
Method
_compute_topk_scores_and_seq
(self, sequences, scores, scores_to_gather, flags)
opennmt-avg/onmt/beam.py:58
↓ 2 callers
Method
_compute_topk_scores_and_seq
(self, sequences, scores, scores_to_gather, flags)
opennmt-baseline/onmt/beam.py:58
↓ 2 callers
Method
_compute_topk_scores_and_seq
(self, sequences, scores, scores_to_gather, flags)
opennmt-self/onmt/beam.py:58
↓ 2 callers
Function
_dataset_loader
(pt_file, corpus_type)
opennmt-feature/inputters/dataset.py:225
↓ 2 callers
Function
_dataset_loader
(pt_file, corpus_type)
opennmt-cnn/inputters/dataset.py:353
↓ 2 callers
Function
_dataset_loader
(pt_file, corpus_type)
opennmt-avg/inputters/dataset.py:353
↓ 2 callers
Function
_dataset_loader
(pt_file, corpus_type)
opennmt-baseline/inputters/dataset.py:211
↓ 2 callers
Function
_dataset_loader
(pt_file, corpus_type)
opennmt-self/inputters/dataset.py:353
↓ 2 callers
Method
_make_shard_state
Make shard state dictionary for shards() to return iterable shards for efficient loss computation. Subclass must define this
opennmt-feature/utils/loss.py:76
↓ 2 callers
Method
_make_shard_state
Make shard state dictionary for shards() to return iterable shards for efficient loss computation. Subclass must define this
opennmt-cnn/utils/loss.py:76
↓ 2 callers
Method
_make_shard_state
Make shard state dictionary for shards() to return iterable shards for efficient loss computation. Subclass must define this
opennmt-avg/utils/loss.py:76
↓ 2 callers
Method
_make_shard_state
Make shard state dictionary for shards() to return iterable shards for efficient loss computation. Subclass must define this
opennmt-baseline/utils/loss.py:76
↓ 2 callers
Method
_make_shard_state
Make shard state dictionary for shards() to return iterable shards for efficient loss computation. Subclass must define this
opennmt-self/utils/loss.py:76
↓ 2 callers
Method
_next_dataset_iterator
(self, dataset_iter)
opennmt-feature/inputters/dataset.py:175
↓ 2 callers
Method
_next_dataset_iterator
(self, dataset_iter)
opennmt-cnn/inputters/dataset.py:303
↓ 2 callers
Method
_next_dataset_iterator
(self, dataset_iter)
opennmt-avg/inputters/dataset.py:303
↓ 2 callers
Method
_next_dataset_iterator
(self, dataset_iter)
opennmt-baseline/inputters/dataset.py:160
↓ 2 callers
Method
_next_dataset_iterator
(self, dataset_iter)
opennmt-self/inputters/dataset.py:303
↓ 2 callers
Function
all_gather_list
Gathers arbitrary data from all nodes into a list.
opennmt-feature/utils/distributed.py:83
↓ 2 callers
Function
all_gather_list
Gathers arbitrary data from all nodes into a list.
opennmt-cnn/utils/distributed.py:83
↓ 2 callers
Function
all_gather_list
Gathers arbitrary data from all nodes into a list.
opennmt-avg/utils/distributed.py:83
↓ 2 callers
Function
all_gather_list
Gathers arbitrary data from all nodes into a list.
opennmt-baseline/utils/distributed.py:89
↓ 2 callers
Function
all_gather_list
Gathers arbitrary data from all nodes into a list.
opennmt-self/utils/distributed.py:83
↓ 2 callers
Method
all_gather_stats
Gather a `Statistics` object accross multiple process/nodes Args: stat(:obj:Statistics): the statistics object to gather
opennmt-feature/utils/statistics.py:30
↓ 2 callers
Method
all_gather_stats
Gather a `Statistics` object accross multiple process/nodes Args: stat(:obj:Statistics): the statistics object to gather
opennmt-cnn/utils/statistics.py:30
↓ 2 callers
Method
all_gather_stats
Gather a `Statistics` object accross multiple process/nodes Args: stat(:obj:Statistics): the statistics object to gather
opennmt-avg/utils/statistics.py:30
↓ 2 callers
Method
all_gather_stats
Gather a `Statistics` object accross multiple process/nodes Args: stat(:obj:Statistics): the statistics object to gather
opennmt-baseline/utils/statistics.py:30
↓ 2 callers
Method
all_gather_stats
Gather a `Statistics` object accross multiple process/nodes Args: stat(:obj:Statistics): the statistics object to gather
opennmt-self/utils/statistics.py:30
↓ 2 callers
Function
all_reduce_and_rescale_tensors
All-reduce and rescale tensors in chunks of the specified size. Args: tensors: list of Tensors to all-reduce rescale_denom: denom
opennmt-feature/utils/distributed.py:31
↓ 2 callers
Function
all_reduce_and_rescale_tensors
All-reduce and rescale tensors in chunks of the specified size. Args: tensors: list of Tensors to all-reduce rescale_denom: denom
opennmt-cnn/utils/distributed.py:31
↓ 2 callers
Function
all_reduce_and_rescale_tensors
All-reduce and rescale tensors in chunks of the specified size. Args: tensors: list of Tensors to all-reduce rescale_denom: denom
opennmt-avg/utils/distributed.py:31
↓ 2 callers
Function
all_reduce_and_rescale_tensors
All-reduce and rescale tensors in chunks of the specified size. Args: tensors: list of Tensors to all-reduce rescale_denom: denom
opennmt-baseline/utils/distributed.py:35
↓ 2 callers
Function
all_reduce_and_rescale_tensors
All-reduce and rescale tensors in chunks of the specified size. Args: tensors: list of Tensors to all-reduce rescale_denom: denom
opennmt-self/utils/distributed.py:31
↓ 2 callers
Function
all_reduce_buffer
()
opennmt-feature/utils/distributed.py:43
↓ 2 callers
Function
all_reduce_buffer
()
opennmt-cnn/utils/distributed.py:43
↓ 2 callers
Function
all_reduce_buffer
()
opennmt-avg/utils/distributed.py:43
↓ 2 callers
Function
all_reduce_buffer
()
opennmt-baseline/utils/distributed.py:49
↓ 2 callers
Function
all_reduce_buffer
()
opennmt-self/utils/distributed.py:43
↓ 2 callers
Function
build_base_model
Args: model_opt: the option loaded from checkpoint. fields: `Field` objects for the model. gpu(bool): whether to use gpu. c
opennmt-feature/onmt/transformer.py:108
↓ 2 callers
Function
build_base_model
Args: model_opt: the option loaded from checkpoint. fields: `Field` objects for the model. gpu(bool): whether to use gpu. c
opennmt-cnn/onmt/transformer.py:106
↓ 2 callers
Function
build_base_model
Args: model_opt: the option loaded from checkpoint. fields: `Field` objects for the model. gpu(bool): whether to use gpu. c
opennmt-avg/onmt/transformer.py:106
↓ 2 callers
Function
build_base_model
Args: model_opt: the option loaded from checkpoint. fields: `Field` objects for the model. gpu(bool): whether to use gpu. c
opennmt-baseline/onmt/transformer.py:100
↓ 2 callers
Function
build_base_model
Args: model_opt: the option loaded from checkpoint. fields: `Field` objects for the model. gpu(bool): whether to use gpu. c
opennmt-self/onmt/transformer.py:106
↓ 2 callers
Function
build_dataset_iter
This returns user-defined train/validate data iterator for the trainer to iterate over. We implement simple ordered iterator strategy here, but
opennmt-feature/inputters/dataset.py:276
↓ 2 callers
Function
build_dataset_iter
This returns user-defined train/validate data iterator for the trainer to iterate over. We implement simple ordered iterator strategy here, but
opennmt-cnn/inputters/dataset.py:451
↓ 2 callers
Function
build_dataset_iter
This returns user-defined train/validate data iterator for the trainer to iterate over. We implement simple ordered iterator strategy here, but
opennmt-avg/inputters/dataset.py:451
↓ 2 callers
Function
build_dataset_iter
This returns user-defined train/validate data iterator for the trainer to iterate over. We implement simple ordered iterator strategy here, but
opennmt-baseline/inputters/dataset.py:248
↓ 2 callers
Function
build_dataset_iter
This returns user-defined train/validate data iterator for the trainer to iterate over. We implement simple ordered iterator strategy here, but
opennmt-self/inputters/dataset.py:451
↓ 2 callers
Function
build_embeddings
Build an Embeddings instance. Args: opt: the option in current environment. word_dict(Vocab): words dictionary. feature_dicts([
opennmt-baseline/onmt/transformer.py:35
↓ 2 callers
Function
build_field_vocab
(field, counter, **kwargs)
opennmt-baseline/preprocess.py:28
↓ 2 callers
Function
build_loss_compute
Returns a LossCompute subclass which wraps around an nn.Module subclass (such as nn.NLLLoss) which defines the loss criterion. The LossCompute
opennmt-feature/utils/loss.py:16
↓ 2 callers
Function
build_loss_compute
Returns a LossCompute subclass which wraps around an nn.Module subclass (such as nn.NLLLoss) which defines the loss criterion. The LossCompute
opennmt-cnn/utils/loss.py:16
↓ 2 callers
Function
build_loss_compute
Returns a LossCompute subclass which wraps around an nn.Module subclass (such as nn.NLLLoss) which defines the loss criterion. The LossCompute
opennmt-avg/utils/loss.py:16
↓ 2 callers
Function
build_loss_compute
Returns a LossCompute subclass which wraps around an nn.Module subclass (such as nn.NLLLoss) which defines the loss criterion. The LossCompute
opennmt-baseline/utils/loss.py:16
↓ 2 callers
Function
build_loss_compute
Returns a LossCompute subclass which wraps around an nn.Module subclass (such as nn.NLLLoss) which defines the loss criterion. The LossCompute
opennmt-self/utils/loss.py:16
↓ 2 callers
Function
build_save_dataset
Building and saving the dataset
opennmt-feature/preprocess.py:251
↓ 2 callers
Function
build_save_dataset
Building and saving the dataset
opennmt-cnn/preprocess.py:457
↓ 2 callers
Function
build_save_dataset
Building and saving the dataset
opennmt-avg/preprocess.py:457
↓ 2 callers
Function
build_save_dataset
Building and saving the dataset
opennmt-baseline/preprocess.py:208
↓ 2 callers
Function
build_save_dataset
Building and saving the dataset
opennmt-self/preprocess.py:457
↓ 2 callers
Method
build_tokens
(self, idx, side="tgt")
opennmt-feature/onmt/translator.py:41
↓ 2 callers
Method
build_tokens
(self, idx, side="tgt")
opennmt-cnn/onmt/translator.py:41
↓ 2 callers
Method
build_tokens
(self, idx, side="tgt")
opennmt-avg/onmt/translator.py:41
↓ 2 callers
Method
build_tokens
(self, idx, side="tgt")
opennmt-baseline/onmt/translator.py:41
↓ 2 callers
Method
build_tokens
(self, idx, side="tgt")
opennmt-self/onmt/translator.py:41
↓ 2 callers
Method
elapsed_time
compute elapsed time
opennmt-feature/utils/statistics.py:99
↓ 2 callers
Method
elapsed_time
compute elapsed time
opennmt-cnn/utils/statistics.py:99
↓ 2 callers
Method
elapsed_time
compute elapsed time
opennmt-avg/utils/statistics.py:99
↓ 2 callers
Method
elapsed_time
compute elapsed time
opennmt-baseline/utils/statistics.py:99
↓ 2 callers
Method
elapsed_time
compute elapsed time
opennmt-self/utils/statistics.py:99
↓ 2 callers
Function
filter_shard_state
(state, shard_size=None)
opennmt-feature/utils/loss.py:241
↓ 2 callers
Function
filter_shard_state
(state, shard_size=None)
opennmt-cnn/utils/loss.py:241
↓ 2 callers
Function
filter_shard_state
(state, shard_size=None)
opennmt-avg/utils/loss.py:241
↓ 2 callers
Function
filter_shard_state
(state, shard_size=None)
opennmt-baseline/utils/loss.py:241
← previous
next →
101–200 of 986, ranked by callers