🤖 Adds initial sim structure

This commit is contained in:
Rune Harlyk
2024-08-09 22:27:48 +02:00
committed by Rune Harlyk
parent 2face72aee
commit 33e7fac74c
23 changed files with 798 additions and 0 deletions
+24
View File
@@ -0,0 +1,24 @@
from simulation.environment import QuadrupedEnv
from training.model import SimpleNN
import resources as resources
def main():
env = QuadrupedEnv(resources.getDataPath() + "/spot.urdf")
env.reset()
input_size = env.robot.get_observation().shape[0]
output_size = env.robot.get_observation().shape[0]
agent = SimpleNN(input_size, output_size)
done = False
observation = []
while not done:
action = agent.select_action(observation)
observation, reward, done = env.step(action)
if __name__ == "__main__":
main()