load trained weights :return: None
(self)
| 153 | tl.files.save_weights_to_hdf5(os.path.join(path, 'pg_policy.hdf5'), self.model) |
| 154 | |
| 155 | def load(self): |
| 156 | """ |
| 157 | load trained weights |
| 158 | :return: None |
| 159 | """ |
| 160 | path = os.path.join('model', '_'.join([ALG_NAME, ENV_ID])) |
| 161 | tl.files.load_hdf5_to_weights_in_order(os.path.join(path, 'pg_policy.hdf5'), self.model) |
| 162 | |
| 163 | |
| 164 | if __name__ == '__main__': |