My Final Project

My project was PyTorch Zoo, a PyTorch library that I made that contains a collection of useful libraries and tools to help reduce the amount of time people spend searching for high-quality implementations of commonly used machine learning modules and utilities.

Link to Code

bkkaggle / pytorch_zoo

A collection of useful modules and utilities (especially helpful for kaggling) not available in Pytorch

Pytorch Zoo

A collection of useful modules and utilities (especially helpful for kaggle) not available in Pytorch

Overview • Installation • Documentation • Contributing • Authors • License • Acknowledgements

Made by Bilal Khan • https://bilal.software

• 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

Installation

pytorch_zoo can be installed from pip

pip install pytorch_zoo

Documentation

Notifications

Sending yourself notifications when your models finish training

IFTTT allows you to easily do this. Follow https://medium.com/datadriveninvestor/monitor-progress-of-your-training-remotely-f9404d71b720 to setup an IFTTT webhook and get a secret key.

Once you have a…

View on GitHub

How I built it (what's the stack? did I run into issues or discover something new along the way?)

I built this project using Python and Pytorch, a commonly used library used for training neural networks. I open-sourced the project on GitHub and you can download it from PyPi by running the command pip install pytorch_zoo