Skip to main content

A collection of small functions that supplements torch functionality

Project description

TorchFun (originally Torchure)

##for those who expect more convenient features from torch

Functionality

  • flatten(tensor)
  • imshow(tensor_batch)
  • load

Install TorchFun

pip install torchfun

API

Flatten (Module)

used to reshape outputs

Usage:

flat = Flatten()
out = flat(x)

flatten (function)

Usage:

out = flatten(x)

imshow (function)

only deal with torch channel-first image batch,

Arguements:

  • x: input data cube, torch tensor or numpy array.
  • title: add title to plot. (Default None)
    • title can be string, or any string-able object.
  • auto_close: (default True)
    • Close the pyplot session afterwards.
    • Clean the environment just like you had never used matplotlib here.
    • if set to False, the plot will remain in the memory for further drawings.

Usage:

imshow(batch) imshow(batch,title=[a,b,c]) imshow(batch,title='title') imshow(batch,auto_close=False)

load (function)

Arguements:

  • arbitrary arguemnts named : a and b Load weight a into model b, or load model b using weight a The order of the arguments doesn't matter. Example:

    load('weights.pts',model)

or

>load(model,'weights.pts')

or

>f = open('weight.pts')
>load(f,model)

or

>load(model,f)

Return value:

  • None

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

torchure-0.0.14.tar.gz (2.9 kB view hashes)

Uploaded source

Built Distribution

torchure-0.0.14-py3-none-any.whl (3.4 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page