Skip to content

Implementation of DenseNet using Tensorflow 2.0 subclassing API

Notifications You must be signed in to change notification settings

vmelan/DenseNet-tf2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation of DenseNet

Implementation of DenseNet using Tensorflow 2.0 subclassing API

model-densenet

Installation

This implementation is based on CIFAR-10 dataset, which can be acquired by the bash file

bash get_data.sh

Concerning TensorFlow, we will need its version 2.0

pip install tensorflow==2.0.0-rc0

Execution

To execute the training phase you will need to run the following command

python main.py

About

Implementation of DenseNet using Tensorflow 2.0 subclassing API

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published