Collection of functions and modules to help development in PyTorch.
MIT License
Bot releases are hidden (Show)
subdir_size
to pack_dataset
.CosDecayScheduler
and CosDecayRule
classes.sorted_dict
to collections.ndim
and shape
to search information in tensor-like objects.item
to convert scalar-like objects to built-in scalar objects.from_numpy
to numpy_to_tensor
to avoid confusion with torch.from_numpy
.save_to_yaml
to to_yaml
and save_to_csv
to to_csv
.PickleDataset
to PackedDataset
.pack_to_pickle
to pack_dataset
.EModule
and ESequential
classes with auto-config and auto-device detection.can_be_converted_to_tensor
now accepts numpy arrays and scalars.from torchoutil import nn
)can_be_stacked
now returns False with an empty sequence.Published by Labbeti 6 months ago
count_parameters
to TModule
.padding_idx
to indices_to_onehot
function.dict_list_to_list_dict
, flat_list
, unflat_dict_of_dict
to collections utils.PickleDataset
and function pack_to_pickle
to utils.ResampleNearest
and function resample_nearest
.TransformDrop
and function transform_drop
.is_scalar
now returns True for numpy scalars when numpy package is installed.Published by Labbeti 6 months ago
PositionalEncoding
layer for transformers networks.metadata
to HDFDataset
.create_params_groups_bias
for avoid apply weight decay to networks bias.is_numpy_scalar
, is_torch_scalar
and unzip
.auto_open
, numpy_to_torch
and file_kwargs
to customize loading in HDFDataset
.at
from HDFDataset
when return_added_columns
is False
.ModelCheckpointRegister
to RegistryHub
and add hash value check after download.can_be_converted_to_tensor
now returns True if input is a Tensor.Published by Labbeti 8 months ago
ModelCheckpointRegister
now creates intermediate directories before download.MaskedMean
and MaskedSum
dim argument.Published by Labbeti 8 months ago
ModelCheckpointRegister
class to make download and loading easier from checkpoint.pack_to_hdf
now supports existing shape column.SizedDatasetLike
is now compatible with Sequence
-like objects.Published by Labbeti 8 months ago
HDFDataset
class and pack_to_hdf
function.diagonal
to generate_square_subsequent_mask
.Abs
, Angle
, Real
and Imag
modules.Published by Labbeti 9 months ago