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
andb
Load weighta
into modelb
, or load modelb
using weighta
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:
- model object
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchfun-0.0.11.tar.gz
(2.9 kB
view hashes)
Built Distribution
Close
Hashes for torchfun-0.0.11-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 22776f0bed2a1d3101f454ba62644f728705eae6d5786ca6e77624f6dbb0376b |
|
MD5 | 19e7d9208a4a307b794c08d3d5fb9dd2 |
|
BLAKE2b-256 | cb78746f6b9bee06dcef8100d39693fafdb27ed499e0e5f2b9e832321e66deb4 |