Skip to content

Latest commit

 

History

History
28 lines (11 loc) · 491 Bytes

File metadata and controls

28 lines (11 loc) · 491 Bytes

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