Algorithm Model

[Yajoker]

Hello, if I want to use my own version of the algorithm to replace yours, how should I do it? Do I just need to add my own version of the algorithm under DRL-robot-navigation-IR-SIM/robot_nav
/models/?

[reiniscimurs]

Hi,

Yes you can add a new model in models directory, then import it in train.py file and initialize it

DRL-robot-navigation-IR-SIM/robot_nav/train.py

Lines 39 to 47 in bf89430

	model = TD3(
	state_dim=state_dim,
	action_dim=action_dim,
	max_action=max_action,
	device=device,
	save_every=save_every,
	load_model=False,
	model_name="TD3",
	) # instantiate a model

Just make sure that the interfaces of your model are the same as base models, or you will have to make some changes to the training cycle or the simulator.

[reiniscimurs]

Of course, I also welcome contributions to this repo. If you have a new model, feel free to open a PR and contribute here.