sample.py

autolearner/model/knowledge/addpostion.py

@@ -0,0 +1,2 @@
+import torch
+import torch.nn as nn

autolearner/model/reason/reasoner.py

@@ -195,6 +195,10 @@ def forward_reason(self, x):
 		return x
 
 class UniversalMap(nn.Module):
+	"""
+	The Univseral Neuro Map that is takes several input channels [d] and output channel [o] as config
+	to create the whole mapping.
+	"""
 	def __init__(self, name, input_dims, output_dim, fc_dim = 64, conv_dim = 64):
 		super().__init__()
 		self.name = name

autolearner/model/reason/sampler.py

@@ -0,0 +1,11 @@
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+
+class TrajectoryDiffuser(nn.Module):
+    """
+    An action diffusion model that is used to perform diffusion and sample valid trajectory
+    for a specific action.
+    """
+    def __init__(self, action_dim, latent_dim, beta = 0.1):
+        super().__init__()