optimizers.h

Go to the documentation of this file.

#pragma once

#include "optimizer/optimizer.h"
#include "optimizer/gradientdescent/gradientdescent.h"

namespace magmadnn {
namespace optimizer {

enum optimizer_t {
    SGD,
    ADAM,
};

enum loss_t {
    CROSS_ENTROPY,
    MSE
};

}   // namespace optimizer
}   // namespace magmadnn