Skip to content
#

Neural Network

Artificial neural networks (ANN) are computational systems that "learn" to perform tasks by considering examples, generally without being programmed with any task-specific rules.

Here are 10,722 public repositories matching this topic...

kurtamohler
kurtamohler commented Jan 20, 2021

🚀 Feature

Add support for torch.max with:

  • CUDA bfloat16
  • CPU float16 and bfloat16

Motivation

Currently, torch.max has support for CUDA float16:

>>> torch.rand(10, dtype=torch.float16, device='cuda').max()
tensor(0.8530, device='cuda:0', dtype=torch.float16)

But all three other combinations of CPU/CUDA and float16/bfloat16 are not supported:

>>> torch.ra
gensim
askhade
askhade commented Jan 19, 2021

Add a new API for converting a model to external data. Today the conversion happens in 2 steps
external_data_helper.convert_model_to_external_data(<model>, <all_tensors_to_one_file>, <size_threshold>) save_model(model, output_path)
We want to add another api which combines the 2 steps
`
save_model_to_external_data(, <output_