reinforcex 0.0.5

Deep Reinforcement Learning Framework
1
2
3
4
5
6
7
8
9

use crate::prob_distributions::BaseDistribution;
use tch::{Device, Tensor};

pub trait BasePolicy {
    fn forward(&self, x: &Tensor) -> (Box<dyn BaseDistribution>, Option<Tensor>);
    fn device(&self) -> Device;
    fn save(&self, path: &str);
    fn load(&mut self, path: &str);
}