reinforce_loss.h

Go to the documentation of this file.

#ifndef REINFORCE_LOSS_H
#define REINFORCE_LOSS_H
 
#include "cuberl/base/cubeai_config.h"
 
#ifdef USE_PYTORCH
 
#include "cuberl/base/cuberl_types.h"
#include "bitrl/bitrl_consts.h"
 
#include <vector>
 
namespace cuberl {
namespace rl {
namespace algos {
namespace pg {
    
    //using namespace bitrl::consts;
    
std::vector<torch_tensor_t>
compute_loss_item(const std::vector<real_t>& rewards, 
                  const std::vector<torch_tensor_t>& log_probs);
 
std::vector<real_t> 
compute_baseline_with_constant(const std::vector<real_t>& rewards,
                               real_t constant);
                               
                               
std::vector<real_t> 
compute_baseline_with_mean(const std::vector<real_t>& rewards);
                           
std::vector<real_t> 
compute_baseline_with_standardization(const std::vector<real_t>& rewards, 
                                      real_t eps=bitrl::consts::TOLERANCE);
    
}
}
}
}
 
 
#endif
#endif