Skip to content

robomechanics/SIDM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 

Repository files navigation

SIDM

Maintainer: Sean J. Wang, [email protected]

Required Dependencies

NumPy

  • Numpy is used for storage of parameters and small math operations

PyTorch

  • PyTorch is used for network training

wheeledRobotSimPybullet

  • wheeledRobotSimPybullet is the simulation used to gather data. This package also contains the dataset class used for the replay buffer. It also contains state transformations used during training.

Dataset

Dataset is generated using a PyBullet simulated wheeled robot. Two different systems are provided. The first has a novice and expert robot that are more similar to each other.

Running SIDM Domain Transfer Learning

After the dataset is generated, execute "sidmTraining.py" to run the SIDM training procedure. "sidmNetworks.py" contains all networks needed for SIDM as well as "sidmLosses" class used to calculate all training losses.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages