diff --git a/benchmark/torch/coma/coma_config.py b/benchmark/torch/coma/coma_config.py index 364ea058f6de2dbe04a9c610b8bb7c89aec9832e..25d6971efe4695f7816cf47148835a1e9d421bc6 100644 --- a/benchmark/torch/coma/coma_config.py +++ b/benchmark/torch/coma/coma_config.py @@ -1,3 +1,17 @@ +# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + # arguments of coma config = { # ========== Environment ========== diff --git a/benchmark/torch/coma/sc2_agent.py b/benchmark/torch/coma/sc2_agent.py index db8ab986468dc061812c5e40ed914841540260a4..08af98521745404d61f0f5c1da30fbefbe9aba9e 100644 --- a/benchmark/torch/coma/sc2_agent.py +++ b/benchmark/torch/coma/sc2_agent.py @@ -1,4 +1,17 @@ -# -*- coding: UTF-8 -*- +# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + import numpy as np import parl import torch diff --git a/benchmark/torch/coma/sc2_model.py b/benchmark/torch/coma/sc2_model.py index abf3d9b7ff49ef432395a673536b6c636265a609..8fd1fa6781e7a62f38cf5f1a9ff8871bbdac5998 100644 --- a/benchmark/torch/coma/sc2_model.py +++ b/benchmark/torch/coma/sc2_model.py @@ -1,4 +1,17 @@ -# -*- coding: UTF-8 -*- +# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + import torch.nn as nn import torch.nn.functional as F import numpy as np diff --git a/benchmark/torch/coma/starcraft2/build_docker.sh b/benchmark/torch/coma/starcraft2/build_docker.sh index 9406390578e4658c7f8dcbb1eea3e2f67b43d5f4..ad7a742e1aa0aa347e7dd9755cf8c20597798292 100644 --- a/benchmark/torch/coma/starcraft2/build_docker.sh +++ b/benchmark/torch/coma/starcraft2/build_docker.sh @@ -1,6 +1,5 @@ #!/bin/bash - #### ------------------------------------------------------------------- #### build docker image #### ------------------------------------------------------------------- diff --git a/benchmark/torch/coma/train.py b/benchmark/torch/coma/train.py index 525c5926527de12d9f04ec2e869ceee7940507f4..6c60b8a072c49a92c7c1edf58fbf034805db1bbb 100644 --- a/benchmark/torch/coma/train.py +++ b/benchmark/torch/coma/train.py @@ -1,3 +1,17 @@ +# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + from smac.env import StarCraft2Env import numpy as np import os diff --git a/parl/algorithms/torch/__init__.py b/parl/algorithms/torch/__init__.py index dc2aabac3dcdd239eba1b059fbf1283a192b4b10..97826766b2581762d1080c05eb979bc4d0e4b03f 100644 --- a/parl/algorithms/torch/__init__.py +++ b/parl/algorithms/torch/__init__.py @@ -16,4 +16,5 @@ from parl.algorithms.torch.ddqn import * from parl.algorithms.torch.dqn import * from parl.algorithms.torch.a2c import * from parl.algorithms.torch.td3 import * +from parl.algorithms.torch.coma import * from parl.algorithms.torch.policy_gradient import *