search

UoA-CARES / cares_reinforcement_learning

CARES Reinforcement Learning Package
11 stars 2 forks source link

Load actor and critic models when create TD3 #80

Closed Bac0nEater closed 1 year ago

Bac0nEater commented 1 year ago

Example of how to pass model file path to the util

actor_file_path = pathlib.Path(ACTOR_PATH).resolve()
critic_file_path = pathlib.Path(CRITIC_PATH).resolve()

network_factory_args["actor_file_path"] = actor_file_path
network_factory_args["critic_file_path"] = critic_file_path