Skip to content

tf.keras implementation to easily launch deep learning models

License

Notifications You must be signed in to change notification settings

medtune/Models_tf_keras

Repository files navigation

Models_tf_keras

Models_tf_keras is a Python library built on top of Tensorflow tensorflow. The starting point of the program is a configuration file, located in "/yaml" folder, under the name of "config.yaml"

Installation

Use the package manager pip to install foobar.

pip install requirement.txt

Usage

dataset:
  dataset_dir: 
  file_pattern: 
  num_samples: 
  num_classes: 
  image_type: 
model:
  name: 
  classification_layers: 
  classification_type: 
  optimizer_noun: 
  activation_func: 
train:
  distribute: 
  xla : 
  batch_size: 
  learning_rate:
    initial : 
    decay_factor : 
    before_decay : 
  num_epochs: 
  shuffle_buffer_size:

Contributing

Coming soon

License

MIT

About

tf.keras implementation to easily launch deep learning models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages