added author info

Author: veds12

a2c/a2c.py

@@ -1,3 +1,6 @@
+# Author : Vedant Shah
+# E-mail : [email protected]
+
 import matplotlib.pyplot as plt
 import torch
 import torch.nn as nn

ddpg/ddpg.py

@@ -1,10 +1,12 @@
+# Author : Vedant Shah
+# E-mail : [email protected]
+
 import copy
 import random
 from collections import deque, namedtuple
 
 import matplotlib.pyplot as plt
 
-import gym
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -302,10 +304,15 @@ def plot(self, plot_type):
 
 
 if __name__ == "__main__":
-    env = gym.make("MountainCarContinuous-v0")
-    env._max_episode_steps = 7000
+    #import gym
+    import pybullet_envs.bullet as bullet
+
+    env = bullet.kukaGymEnv.KukaGymEnv(renders=False, isDiscrete=False)
+    #env = gym.make("MountainCarContinuous-v0")
+    #env._max_episode_steps = 7000
     myagent = agent(env)
-    myagent.train(max_episodes=75)
+    myagent.train()
     myagent.plot("train")
+
     # myagent.test()
     # myagent.plot("test")

dqn/dqn.py

@@ -1,4 +1,7 @@
-import math
+# Author : Vedant Shah
+# E-mail : [email protected]
+
+ math
 import random
 from collections import deque, namedtuple
 

gcn/layers.py

@@ -6,6 +6,8 @@
 Original file is located at
     https://colab.research.google.com/drive/1fCQ_zLCcWNzgE99LK9B2cWrql8J3HgBO
 """
+# Author : Vedant Shah
+# E-mail : [email protected]
 
 import torch
 import torch.nn as nn

gcn/model.py

@@ -1,3 +1,6 @@
+# Author : Vedant Shah
+# E-mail : [email protected]
+
 import torch.nn as nn
 import torch.nn.functional as F
 

q-learning/q_learning.py

@@ -1,3 +1,6 @@
+# Author : Vedant Shah
+# E-mail : [email protected]
+
 import gym
 import numpy as np
 

vpg/vpg.py

@@ -1,3 +1,6 @@
+# Author : Vedant Shah
+# E-mail : [email protected]
+
 import gym
 import torch
 from torch.distributions.categorical import Categorical