search

ssmmoo1 / RL_taskset_scheduling

Trains an RL agent based on a graph neural network to schedule real time periodic task sets on a set of heterogeneous processors.
3 stars 1 forks source link

Implement REINFORCE #7

Closed ssmmoo1 closed 1 year ago

ssmmoo1 commented 2 years ago

Need to implement the REINFORCE algorithm to train the GNN.

ssmmoo1 commented 1 year ago

Basic REINFORCE code is added in main. Need to test and debug.

ssmmoo1 commented 1 year ago

Done it works.