专栏首页深度学习与数据挖掘实战干货|PyTorch实用代码段集锦

干货|PyTorch实用代码段集锦

Facebook公司开源的深度学习框架PyTorch越来越火,PyTorch易于上手。本文节选github中PyTorch的常用实用代码片段,供大家学习参考。

Github链接地址

https://github.com/ptrblck/pytorch_misc

代码总览

  • accumulate_gradients - Comparison of accumulated gradients/losses to vanilla batch update. 链接:https://github.com/ptrblck/pytorch_misc/blob/master/accumulate_gradients.py
  • adaptive_batchnorm- Adaptive BN implementation using two additional parameters: out = a * x + b * bn(x). 链接: https://github.com/ptrblck/pytorch_misc/blob/master/adaptive_batchnorm.py
  • adaptive_pooling_torchvision - Example of using adaptive pooling layers in pretrained models to use different spatial input shapes. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/adaptive_pooling_torchvision.py
  • batch_norm_manual - Comparison of PyTorch BatchNorm layers and a manual calculation. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/batch_norm_manual.py
  • change_crop_in_dataset - Change the image crop size on the fly using a Dataset. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/change_crop_in_dataset.py
  • conv_rnn - Combines a 3DCNN with an RNN; uses windowed frames as inputs. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/conv_rnn.py
  • csv_chunk_read - Provide data chunks from continuous .csv file. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/csv_chunk_read.py
  • densenet_forwardhook - Use forward hooks to get intermediate activations from densenet121. Uses separate modules to process these activations further. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/densenet_forwardhook.py
  • edge_weighting_segmentation - Apply weighting to edges for a segmentation task. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/edge_weighting_segmentation.py
  • image_rotation_with_matrix - Rotate an image given an angle using 1.) a nested loop and 2.) a rotation matrix and mesh grid. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/image_rotation_with_matrix.py
  • LocallyConnected2d - Implementation of a locally connected 2d layer. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/LocallyConnected2d.py
  • mnist_autoencoder - Simple autoencoder for MNIST data. Includes visualizations of output images, intermediate activations and conv kernels. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/mnist_autoencoder.py
  • mnist_permuted - MNIST training using permuted pixel locations. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/mnist_permuted.py
  • model_sharding_data_parallel - Model sharding with DataParallel using 2 pairs of 2 GPUs. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/model_sharding_data_parallel.py
  • momentum_update_nograd - Script to see how parameters are updated when an optimizer is used with momentum/running estimates, even if gradients are zero. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/momentum_update_nograd.py
  • shared_array - Script to demonstrate the usage of shared arrays using multiple workers. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/shared_array.py
  • unet_demo - Simple UNet demo. 链接: https://github.com/ptrblck/pytorch_misc/blob/master/unet_demo.py
  • weighted_sampling - Usage of WeightedRandomSampler using an imbalanced dataset with class imbalance 99 to 1 链接: https://github.com/ptrblck/pytorch_misc/blob/master/weighted_sampling.py

本文分享自微信公众号 - 深度学习与数据挖掘实战(www_datageekers_com),作者:数据极客社区

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-11-19

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 干货|RSNA肺炎检测挑战赛冠军经验分享

          在国际上,肺炎占所有5岁以下儿童死亡的15%以上。2015年,有920,000名5岁以下儿童死于该病。在美国,肺炎占急诊科就诊人数超过50万[1],...

    fishexpert
  • 干货|RSNA肺炎检测挑战赛冠军经验分享

    在国际上,肺炎占所有5岁以下儿童死亡的15%以上。2015年,有920,000名5岁以下儿童死于该病。在美国,肺炎占急诊科就诊人数超过50万[1],...

    fishexpert
  • 【社交网络挖掘&Jupyter&贝叶斯】资源分享

    https://www.packtpub.com/books/content/getting-started-jupyter-notebook-part-1

    fishexpert
  • NLP之keras中文文本分类系列算法封装,简单易用(超详细教程)

    中文长文本分类、短句子分类、多标签分类、两句子相似度(Chinese Text Classification of Keras NLP, multi-label...

    大数据技术与机器学习
  • Spark Shuffle数据处理过程与部分调优(源码阅读七)

      shuffle。。。相当重要,为什么咩,因为shuffle的性能优劣直接决定了整个计算引擎的性能和吞吐量。相比于Hadoop的MapReduce,可以看到S...

    用户3003813
  • NLP简报

    欢迎来到 NLP 时事简报!涵盖了诸如特定语言 BERT 模型、免费数据集、深度学习库等主题。

    NewBeeNLP
  • AIoT=AI+IoT,背后究竟是谁更需要谁?

    自去年年初,伴随着BAT、小米、华为、京东、旷视等相继喊出AIoT的口号,这一名词以及其所代表的背后产业引发新一轮行业热议。

    镁客网
  • 新手如何发布第一个Python项目开源包?这里有一份详细指南

    作者以 SciTime 项目(一个对算法训练时间进行估计的包)的发布为例,详细解释了发布的每个步骤。

    机器之心
  • 新手如何发布第一个Python项目开源包?这里有一份详细指南

    作者以 SciTime 项目(一个对算法训练时间进行估计的包)的发布为例,详细解释了发布的每个步骤。

    CDA数据分析师
  • 在线教育风口已来,这几个开源项目不能错过 | Gitee 项目推荐

    由于疫情严重,全国学生全部推迟了开学时间,但是停课不停学,许多学校都已经开始了线上课程。对于线上教育行业来说,这也是一次发展壮大的机会。线上教育想要积累稳定的客...

    后端码匠

扫码关注云+社区

领取腾讯云代金券