Skip to content

Latest commit

 

History

History

pytorch

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
demo_kaf_convolutional.py
demo_kaf_convolutional.py
 
 
demo_kaf_feedforward.py
demo_kaf_feedforward.py
 
 
kafnets.py
kafnets.py
 
 

README.md

Kernel activation functions (PyTorch)

In the kafnets module you can find the modules for defining KAF layers, both for feedforward networks and convolutional networks (using the flag 'conv' during initialization). The code has two demos to showcase the modules using the PyTorch sequential class.

Requirements

  • pytorch = 1.0.1
  • numpy = 1.15.4
  • tqdm = 4.28.1 (for demo_kaf_convolutional.py)
  • scikit-learn = 0.20.1 (for demo_feedforward.py)