EASY CS Technology

Home
Forum
Members
More

preview-blocker.popup

CS Forum

Artificial Intelligence

PyTorch Zoo：PyTorch模块/程序集

Search

May 14, 2020

PyTorch Zoo：PyTorch模块/程序集

Installation
Documentation
Notifications
Sending yourself notifications when your models finish training
Viewing training progress with tensorboard in a kaggle kernel
Data
DynamicSampler(sampler, batch_size=32)
trim_tensors(tensors)
Loss
lovasz_hinge(logits, labels, per_image=True)
DiceLoss()
Metrics
Modules
SqueezeAndExcitation(in_ch, r=16)
ChannelSqueezeAndSpatialExcitation(in_ch)
ConcurrentSpatialAndChannelSqueezeAndChannelExcitation(in_ch)
GaussianNoise(0.1)
Schedulers
CyclicalMomentum(optimizer, base_momentum=0.8, max_momentum=0.9, step_size=2000, mode="triangular")
Utils
notify({'value1': 'Notification title', 'value2': 'Notification body'}, key)
seed_environment(seed=42)
gpu_usage(device, digits=4)
n_params(model)
save_model(model, fold=0)
load_model(model, fold=0)
save(obj, 'obj.pkl')
load('obj.pkl')
masked_softmax(logits, mask, dim=-1)
masked_log_softmax(logits, mask, dim=-1)
Contributing
Authors
License
Acknowledgements

github地址：https://github.com/bkkaggle/pytorch_zoo

0 comments

0

©2020 by EasyCSTech. Special thanks to IPinfo and EasyNote.