crypt3lx2k

#! /usr/bin/env python

import tensorflow as tf

def inference_fn(x, training=False):
    net = x

    net = tf.layers.flatten(net)
    net = tf.layers.dense(net, 512, activation=tf.nn.relu)
    net = tf.layers.dropout(net, 0.2, training=training)

jeremyjordan

from keras.callbacks import Callback
import keras.backend as K
import numpy as np

class SGDRScheduler(Callback):
    '''Cosine annealing learning rate scheduler with periodic restarts.

    # Usage
        ```python
            schedule = SGDRScheduler(min_lr=1e-5,

alsrgv

from __future__ import print_function

import os
import sys
import tensorflow as tf
from google.protobuf import text_format
from tensorflow.python.framework import graph_io

if len(sys.argv) < 2:
    print('Usage: %s <filename prefix>' % sys.argv[0])