Lotta work here
This commit is contained in:
1 parent
d83d07a823
commit
d89095e49b
34 files changed
+3207
-186
No files matched your search
@@ -0,0 +1,20 @@
|
||||
#pragma once
|
||||
|
||||
#include "../core/tensor.hpp"
|
||||
#include <vector>
|
||||
|
||||
namespace lm {
|
||||
|
||||
class AdamOptimizer {
|
||||
public:
|
||||
AdamOptimizer(float learning_rate = 0.001, float beta1 = 0.9, float beta2 = 0.999, float epsilon = 1e-8);
|
||||
void step(std::vector<Tensor>& parameters); // Remove const
|
||||
void zero_grad(std::vector<Tensor>& parameters); // Remove const
|
||||
|
||||
private:
|
||||
float learning_rate_, beta1_, beta2_, epsilon_;
|
||||
int timestep_;
|
||||
std::vector<Tensor> m_, v_; // First and second moment estimates
|
||||
};
|
||||
|
||||
} // namespace lm
|
||||
Reference in new issue
Block a user