from stable_baselines3.common.policies import register_policy
from NerveNet.policies.a2c_policies import ActorCriticGnnPolicy, ActorCriticGnnPolicy_V0, ActorCriticGnnPolicy_V2,  ActorCriticGNNPolicyTransfer, ActorCriticMLPPolicyTransfer

register_policy("GnnPolicy", ActorCriticGnnPolicy)
register_policy("GnnPolicy_V0", ActorCriticGnnPolicy_V0)
register_policy("GnnPolicy_V2", ActorCriticGnnPolicy_V2)
register_policy("MLPTransferPolicy", ActorCriticMLPPolicyTransfer)
register_policy("GNNTransferPolicy", ActorCriticGNNPolicyTransfer)

640ca1dbe78ea688cd149af0bc3f99997ff7eb50
<mask token>

<mask token>
register_policy('GnnPolicy', ActorCriticGnnPolicy)
register_policy('GnnPolicy_V0', ActorCriticGnnPolicy_V0)
register_policy('GnnPolicy_V2', ActorCriticGnnPolicy_V2)
register_policy('MLPTransferPolicy', ActorCriticMLPPolicyTransfer)
register_policy('GNNTransferPolicy', ActorCriticGNNPolicyTransfer)

from stable_baselines3.common.policies import register_policy
from NerveNet.policies.a2c_policies import ActorCriticGnnPolicy, ActorCriticGnnPolicy_V0, ActorCriticGnnPolicy_V2, ActorCriticGNNPolicyTransfer, ActorCriticMLPPolicyTransfer
register_policy('GnnPolicy', ActorCriticGnnPolicy)
register_policy('GnnPolicy_V0', ActorCriticGnnPolicy_V0)
register_policy('GnnPolicy_V2', ActorCriticGnnPolicy_V2)
register_policy('MLPTransferPolicy', ActorCriticMLPPolicyTransfer)
register_policy('GNNTransferPolicy', ActorCriticGNNPolicyTransfer)

from stable_baselines3.common.policies import register_policy
from NerveNet.policies.a2c_policies import ActorCriticGnnPolicy, ActorCriticGnnPolicy_V0, ActorCriticGnnPolicy_V2,  ActorCriticGNNPolicyTransfer, ActorCriticMLPPolicyTransfer

register_policy("GnnPolicy", ActorCriticGnnPolicy)
register_policy("GnnPolicy_V0", ActorCriticGnnPolicy_V0)
register_policy("GnnPolicy_V2", ActorCriticGnnPolicy_V2)
register_policy("MLPTransferPolicy", ActorCriticMLPPolicyTransfer)
register_policy("GNNTransferPolicy", ActorCriticGNNPolicyTransfer)
