Skip to content

Some basic utilities for working with PyTorch and torms3's data_provider

License

Notifications You must be signed in to change notification settings

nicholasturner1/PyTorchUtils

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

98 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorchUtils

Some utilities for working with PyTorch and torms3's DataProvider for connectomics data

This package is meant to be EXTREMELY lightweight. It represents a notetaking system rather than a stable updated codebase. This decision was motivated by observing that different experiments often require tweaks that make clean code abstraction difficult. Instead, this tries to abstract a core set of functions (training, inference), and record everything else as separate files. These files are often extremely redundant, but they also allow us to keep precise records and replicate experiments without difficulty.

Required Packages
PyTorch
DataProvider
numpy
h5py
tensorboardX

About

Some basic utilities for working with PyTorch and torms3's data_provider

Resources

License

Stars

Watchers

Forks

Languages