From 42374798b0fd27ad67b41d64f5bce0015503f8e2 Mon Sep 17 00:00:00 2001
From: MilagrosMarin <milmarina02@gmail.com>
Date: Sat, 17 Aug 2024 01:24:06 +0100
Subject: [PATCH] update images according to design reverted

---
 images/pipeline.svg             | 346 ++++++++++++++++----------------
 images/pipeline_moseq_infer.svg | 124 ++++++------
 images/pipeline_moseq_train.svg | 296 +++++++++------------------
 3 files changed, 327 insertions(+), 439 deletions(-)

diff --git a/images/pipeline.svg b/images/pipeline.svg
index 57584ad..a1978fe 100644
--- a/images/pipeline.svg
+++ b/images/pipeline.svg
@@ -1,266 +1,266 @@
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="922pt" height="519pt" viewBox="0.00 0.00 922.09 519.20">
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="934pt" height="519pt" viewBox="0.00 0.00 934.09 519.20">
 <g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 515.2)">
-<polygon fill="white" stroke="none" points="-4,4 -4,-515.2 918.09,-515.2 918.09,4 -4,4"/>
-<!-- moseq_train.LatentDimension -->
+<polygon fill="white" stroke="none" points="-4,4 -4,-515.2 930.09,-515.2 930.09,4 -4,4"/>
+<!-- moseq_infer.Inference.GridMoviesSampledInstances -->
 <g id="node1" class="node">
-<title>moseq_train.LatentDimension</title>
-<g id="a_node1"><a xlink:title="→ moseq_train.PCAFit
------------------------------
variance_percentage  
latent_dimension     
latent_dim_desc      
">
-<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="114.84" cy="-79.2" rx="114.84" ry="17.28"/>
-<text text-anchor="middle" x="114.84" y="-74.92" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.LatentDimension</text>
+<title>moseq_infer.Inference.GridMoviesSampledInstances</title>
+<g id="a_node1"><a xlink:title="→ moseq_infer.Inference
syllable             
------------------------------
instances            
">
+<polygon fill="none" stroke="none" points="718.97,-292.24 472.72,-292.24 472.72,-272.24 718.97,-272.24 718.97,-292.24"/>
+<text text-anchor="start" x="480.72" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.GridMoviesSampledInstances</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.Inference -->
+<!-- moseq_train.PreFit -->
 <g id="node2" class="node">
-<title>moseq_infer.Inference</title>
-<g id="a_node2"><a xlink:title="→ moseq_infer.InferenceTask
------------------------------
inference_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="651.84" cy="-352.8" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="651.84" y="-348.53" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_infer.Inference</text>
+<title>moseq_train.PreFit</title>
+<g id="a_node2"><a xlink:title="→ moseq_train.PreFitTask
------------------------------
model_name=&quot;&quot;        
pre_fit_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="480.84" cy="-12.96" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="480.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PreFit</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.Inference.GridMoviesSampledInstances -->
-<g id="node14" class="node">
-<title>moseq_infer.Inference.GridMoviesSampledInstances</title>
-<g id="a_node14"><a xlink:title="→ moseq_infer.Inference
syllable             
------------------------------
instances            
">
-<polygon fill="none" stroke="none" points="706.97,-292.24 460.72,-292.24 460.72,-272.24 706.97,-272.24 706.97,-292.24"/>
-<text text-anchor="start" x="468.72" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.GridMoviesSampledInstances</text>
+<!-- moseq_train.PCAPrep -->
+<g id="node3" class="node">
+<title>moseq_train.PCAPrep</title>
+<g id="a_node3"><a xlink:title="→ moseq_train.PCATask
------------------------------
coordinates          
confidences          
formatted_bodyparts  
average_frame_rate   
frame_rates          
">
+<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="319.84" cy="-211.68" rx="88.87" ry="17.28"/>
+<text text-anchor="middle" x="319.84" y="-207.41" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.PCAPrep</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.GridMoviesSampledInstances -->
+<!-- moseq_train.PCAFit -->
+<g id="node7" class="node">
+<title>moseq_train.PCAFit</title>
+<g id="a_node7"><a xlink:title="→ moseq_train.PCAPrep
------------------------------
pca_fit_time=null    
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="319.84" cy="-145.44" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="319.84" y="-141.16" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PCAFit</text>
+</a>
+</g>
+</g>
+<!-- moseq_train.PCAPrep&#45;&gt;moseq_train.PCAFit -->
 <g id="edge1" class="edge">
-<title>moseq_infer.Inference-&gt;moseq_infer.Inference.GridMoviesSampledInstances</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M643.23,-343.12C630.22,-330 605.69,-305.27 592.6,-292.07"/>
+<title>moseq_train.PCAPrep-&gt;moseq_train.PCAFit</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-194.14C319.84,-182.84 319.84,-168.14 319.84,-158.06"/>
 </g>
-<!-- moseq_infer.Inference.MotionSequence -->
-<g id="node18" class="node">
-<title>moseq_infer.Inference.MotionSequence</title>
-<g id="a_node18"><a xlink:title="→ moseq_infer.Inference
video_name           
------------------------------
syllable             
latent_state         
centroid             
heading              
">
-<polygon fill="none" stroke="none" points="913.09,-292.24 724.59,-292.24 724.59,-272.24 913.09,-272.24 913.09,-292.24"/>
-<text text-anchor="start" x="732.59" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.MotionSequence</text>
+<!-- moseq_train.LatentDimension -->
+<g id="node4" class="node">
+<title>moseq_train.LatentDimension</title>
+<g id="a_node4"><a xlink:title="→ moseq_train.PCAFit
------------------------------
variance_percentage  
latent_dimension     
latent_dim_desc      
">
+<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="114.84" cy="-79.2" rx="114.84" ry="17.28"/>
+<text text-anchor="middle" x="114.84" y="-74.92" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.LatentDimension</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.MotionSequence -->
+<!-- moseq_infer.VideoRecording.File -->
+<g id="node5" class="node">
+<title>moseq_infer.VideoRecording.File</title>
+<g id="a_node5"><a xlink:title="→ moseq_infer.VideoRecording
file_id              
------------------------------
file_path            
">
+<polygon fill="none" stroke="none" points="920.84,-433.36 760.84,-433.36 760.84,-413.36 920.84,-413.36 920.84,-433.36"/>
+<text text-anchor="start" x="768.84" y="-420.86" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.VideoRecording.File</text>
+</a>
+</g>
+</g>
+<!-- moseq_train.KeypointSet -->
+<g id="node6" class="node">
+<title>moseq_train.KeypointSet</title>
+<g id="a_node6"><a xlink:title="kpset_id             
------------------------------
→ moseq_infer.PoseEstimationMethod
kpset_dir            
kpset_desc=&quot;&quot;        
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="566.97,-440.64 416.72,-440.64 416.72,-406.08 566.97,-406.08 566.97,-440.64"/>
+<text text-anchor="start" x="424.72" y="-420.09" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.KeypointSet</text>
+</a>
+</g>
+</g>
+<!-- moseq_train.Bodyparts -->
+<g id="node10" class="node">
+<title>moseq_train.Bodyparts</title>
+<g id="a_node10"><a xlink:title="→ moseq_train.KeypointSet
bodyparts_id         
------------------------------
anterior_bodyparts   
posterior_bodyparts  
use_bodyparts        
bodyparts_desc=&quot;&quot;    
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="389.34,-370.08 250.34,-370.08 250.34,-335.52 389.34,-335.52 389.34,-370.08"/>
+<text text-anchor="start" x="258.34" y="-349.53" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.Bodyparts</text>
+</a>
+</g>
+</g>
+<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.Bodyparts -->
 <g id="edge2" class="edge">
-<title>moseq_infer.Inference-&gt;moseq_infer.Inference.MotionSequence</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M663.32,-347.09C690.88,-335.77 761.16,-306.92 797.04,-292.19"/>
+<title>moseq_train.KeypointSet-&gt;moseq_train.Bodyparts</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M449.77,-405.59C423.01,-394.92 388.51,-381.17 361.78,-370.52"/>
 </g>
-<!-- moseq_train.PCAFit -->
-<g id="node3" class="node">
-<title>moseq_train.PCAFit</title>
-<g id="a_node3"><a xlink:title="→ moseq_train.PCAPrep
------------------------------
pca_fit_time=null    
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="307.84" cy="-145.44" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="307.84" y="-141.16" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PCAFit</text>
+<!-- moseq_train.KeypointSet.VideoFile -->
+<g id="node17" class="node">
+<title>moseq_train.KeypointSet.VideoFile</title>
+<g id="a_node17"><a xlink:title="→ moseq_train.KeypointSet
video_id             
------------------------------
video_path           
">
+<polygon fill="none" stroke="none" points="576.34,-362.8 407.34,-362.8 407.34,-342.8 576.34,-342.8 576.34,-362.8"/>
+<text text-anchor="start" x="415.34" y="-350.3" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.KeypointSet.VideoFile</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.PCAFit&#45;&gt;moseq_train.LatentDimension -->
+<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.KeypointSet.VideoFile -->
 <g id="edge3" class="edge">
+<title>moseq_train.KeypointSet-&gt;moseq_train.KeypointSet.VideoFile</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M491.84,-405.76C491.84,-392.13 491.84,-373.36 491.84,-362.51"/>
+</g>
+<!-- moseq_train.PCAFit&#45;&gt;moseq_train.LatentDimension -->
+<g id="edge4" class="edge">
 <title>moseq_train.PCAFit-&gt;moseq_train.LatentDimension</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M296.06,-140.52C269.79,-131.77 205.23,-110.28 160.71,-95.47"/>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.93,-140.71C280.33,-132.06 210.71,-110.24 163.07,-95.31"/>
 </g>
 <!-- moseq_train.FullFitTask -->
-<g id="node10" class="node">
+<g id="node11" class="node">
 <title>moseq_train.FullFitTask</title>
-<g id="a_node10"><a xlink:title="→ moseq_train.PCAFit
fullfit_id           
------------------------------
full_latent_dim      
full_kappa           
full_num_iterations  
model_name           
task_mode=&quot;load&quot;     
full_fit_desc=&quot;&quot;     
">
-<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="367.97,-96.48 247.72,-96.48 247.72,-61.92 367.97,-61.92 367.97,-96.48"/>
-<text text-anchor="start" x="255.72" y="-76.7" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.FullFitTask</text>
+<g id="a_node11"><a xlink:title="→ moseq_train.PCAFit
full_latent_dim      
full_kappa           
full_num_iterations  
------------------------------
model_name           
task_mode=&quot;load&quot;     
full_fit_desc=&quot;&quot;     
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="391.59,-96.48 248.09,-96.48 248.09,-61.92 391.59,-61.92 391.59,-96.48"/>
+<text text-anchor="start" x="256.09" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.FullFitTask</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.PCAFit&#45;&gt;moseq_train.FullFitTask -->
-<g id="edge4" class="edge">
+<g id="edge5" class="edge">
 <title>moseq_train.PCAFit-&gt;moseq_train.FullFitTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M307.84,-132.66C307.84,-122.37 307.84,-107.36 307.84,-96.02"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M319.84,-132.66C319.84,-122.52 319.84,-107.78 319.84,-96.5"/>
 </g>
 <!-- moseq_train.PreFitTask -->
-<g id="node11" class="node">
+<g id="node16" class="node">
 <title>moseq_train.PreFitTask</title>
-<g id="a_node11"><a xlink:title="→ moseq_train.PCAFit
prefit_id            
------------------------------
pre_latent_dim       
pre_kappa            
pre_num_iterations   
model_name           
task_mode=&quot;load&quot;     
pre_fit_desc=&quot;&quot;      
">
-<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="505.59,-96.48 386.09,-96.48 386.09,-61.92 505.59,-61.92 505.59,-96.48"/>
-<text text-anchor="start" x="394.09" y="-76.7" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.PreFitTask</text>
+<g id="a_node16"><a xlink:title="→ moseq_train.PCAFit
pre_latent_dim       
pre_kappa            
pre_num_iterations   
------------------------------
model_name           
task_mode=&quot;load&quot;     
pre_fit_desc=&quot;&quot;      
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="551.84,-96.48 409.84,-96.48 409.84,-61.92 551.84,-61.92 551.84,-96.48"/>
+<text text-anchor="start" x="417.84" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.PreFitTask</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.PCAFit&#45;&gt;moseq_train.PreFitTask -->
-<g id="edge5" class="edge">
+<g id="edge6" class="edge">
 <title>moseq_train.PCAFit-&gt;moseq_train.PreFitTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M319.13,-139.18C338.96,-129.95 380.79,-110.48 410.97,-96.43"/>
-</g>
-<!-- moseq_infer.Model -->
-<g id="node4" class="node">
-<title>moseq_infer.Model</title>
-<g id="a_node4"><a xlink:title="model_id             
------------------------------
model_name           
model_dir            
latent_dim           
kappa                
model_desc=&quot;&quot;        
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="710.47,-511.2 593.22,-511.2 593.22,-476.64 710.47,-476.64 710.47,-511.2"/>
-<text text-anchor="start" x="601.22" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.Model</text>
-</a>
-</g>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M331.42,-139.82C353.55,-130.99 402.87,-111.31 438.81,-96.97"/>
 </g>
 <!-- moseq_infer.InferenceTask -->
-<g id="node16" class="node">
+<g id="node8" class="node">
 <title>moseq_infer.InferenceTask</title>
-<g id="a_node16"><a xlink:title="→ moseq_infer.VideoRecording
→ moseq_infer.Model
------------------------------
→ moseq_infer.PoseEstimationMethod
keypointset_dir      
inference_output_dir=&quot;&quot; 
inference_desc=&quot;&quot;    
num_iterations=null  
task_mode=&quot;load&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="731.09,-440.64 572.59,-440.64 572.59,-406.08 731.09,-406.08 731.09,-440.64"/>
-<text text-anchor="middle" x="651.84" y="-419.09" font-family="arial" font-size="12.00" fill="darkgreen">moseq_infer.InferenceTask</text>
-</a>
-</g>
-</g>
-<!-- moseq_infer.Model&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge6" class="edge">
-<title>moseq_infer.Model-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M651.84,-476.32C651.84,-465.54 651.84,-451.55 651.84,-440.8"/>
-</g>
-<!-- moseq_infer.VideoRecording -->
-<g id="node5" class="node">
-<title>moseq_infer.VideoRecording</title>
-<g id="a_node5"><a xlink:title="→ session.Session
recording_id         
------------------------------
→ lab.Device
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="914.09,-511.2 743.59,-511.2 743.59,-476.64 914.09,-476.64 914.09,-511.2"/>
-<text text-anchor="start" x="751.59" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.VideoRecording</text>
+<g id="a_node8"><a xlink:title="→ moseq_infer.VideoRecording
→ moseq_infer.Model
------------------------------
→ moseq_infer.PoseEstimationMethod
keypointset_dir      
inference_output_dir=&quot;&quot; 
inference_desc=&quot;&quot;    
num_iterations=null  
task_mode=&quot;load&quot;     
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="743.09,-440.64 584.59,-440.64 584.59,-406.08 743.09,-406.08 743.09,-440.64"/>
+<text text-anchor="middle" x="663.84" y="-419.09" font-family="arial" font-size="12.00" fill="darkgreen">moseq_infer.InferenceTask</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.VideoRecording.File -->
-<g id="node13" class="node">
-<title>moseq_infer.VideoRecording.File</title>
-<g id="a_node13"><a xlink:title="→ moseq_infer.VideoRecording
file_id              
------------------------------
file_path            
">
-<polygon fill="none" stroke="none" points="908.84,-433.36 748.84,-433.36 748.84,-413.36 908.84,-413.36 908.84,-433.36"/>
-<text text-anchor="start" x="756.84" y="-420.86" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.VideoRecording.File</text>
+<!-- moseq_infer.Inference -->
+<g id="node18" class="node">
+<title>moseq_infer.Inference</title>
+<g id="a_node18"><a xlink:title="→ moseq_infer.InferenceTask
------------------------------
inference_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="663.84" cy="-352.8" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="663.84" y="-348.53" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_infer.Inference</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.VideoRecording.File -->
+<!-- moseq_infer.InferenceTask&#45;&gt;moseq_infer.Inference -->
 <g id="edge7" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.VideoRecording.File</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M828.84,-476.32C828.84,-462.69 828.84,-443.92 828.84,-433.07"/>
-</g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge8" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M785.54,-476.15C758.01,-465.48 722.5,-451.73 695,-441.08"/>
+<title>moseq_infer.InferenceTask-&gt;moseq_infer.Inference</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M663.84,-405.76C663.84,-393.4 663.84,-376.81 663.84,-365.73"/>
 </g>
 <!-- moseq_train.PCATask -->
-<g id="node6" class="node">
+<g id="node9" class="node">
 <title>moseq_train.PCATask</title>
-<g id="a_node6"><a xlink:title="→ moseq_train.Bodyparts
------------------------------
kpms_project_output_dir=&quot;&quot; 
task_mode=&quot;load&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="374.72,-299.52 240.97,-299.52 240.97,-264.96 374.72,-264.96 374.72,-299.52"/>
-<text text-anchor="middle" x="307.84" y="-277.97" font-family="arial" font-size="12.00" fill="darkgreen">moseq_train.PCATask</text>
-</a>
-</g>
-</g>
-<!-- moseq_train.PCAPrep -->
-<g id="node15" class="node">
-<title>moseq_train.PCAPrep</title>
-<g id="a_node15"><a xlink:title="→ moseq_train.PCATask
------------------------------
coordinates          
confidences          
formatted_bodyparts  
average_frame_rate   
frame_rates          
">
-<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="307.84" cy="-211.68" rx="88.87" ry="17.28"/>
-<text text-anchor="middle" x="307.84" y="-207.41" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.PCAPrep</text>
+<g id="a_node9"><a xlink:title="→ moseq_train.Bodyparts
------------------------------
kpms_project_output_dir=&quot;&quot; 
task_mode=&quot;load&quot;     
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="386.72,-299.52 252.97,-299.52 252.97,-264.96 386.72,-264.96 386.72,-299.52"/>
+<text text-anchor="middle" x="319.84" y="-277.97" font-family="arial" font-size="12.00" fill="darkgreen">moseq_train.PCATask</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.PCATask&#45;&gt;moseq_train.PCAPrep -->
-<g id="edge9" class="edge">
+<g id="edge8" class="edge">
 <title>moseq_train.PCATask-&gt;moseq_train.PCAPrep</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.84,-264.64C307.84,-253.86 307.84,-239.87 307.84,-229.12"/>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-264.64C319.84,-253.86 319.84,-239.87 319.84,-229.12"/>
+</g>
+<!-- moseq_train.Bodyparts&#45;&gt;moseq_train.PCATask -->
+<g id="edge9" class="edge">
+<title>moseq_train.Bodyparts-&gt;moseq_train.PCATask</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-335.2C319.84,-324.42 319.84,-310.43 319.84,-299.68"/>
 </g>
 <!-- moseq_train.FullFit -->
-<g id="node7" class="node">
+<g id="node13" class="node">
 <title>moseq_train.FullFit</title>
-<g id="a_node7"><a xlink:title="→ moseq_train.FullFitTask
------------------------------
model_name           
full_fit_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="307.84" cy="-12.96" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="307.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.FullFit</text>
+<g id="a_node13"><a xlink:title="→ moseq_train.FullFitTask
------------------------------
model_name           
full_fit_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="319.84" cy="-12.96" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="319.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.FullFit</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.KeypointSet -->
-<g id="node8" class="node">
-<title>moseq_train.KeypointSet</title>
-<g id="a_node8"><a xlink:title="kpset_id             
------------------------------
→ moseq_infer.PoseEstimationMethod
kpset_dir            
kpset_desc=&quot;&quot;        
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="554.97,-440.64 404.72,-440.64 404.72,-406.08 554.97,-406.08 554.97,-440.64"/>
-<text text-anchor="start" x="412.72" y="-420.09" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.KeypointSet</text>
-</a>
-</g>
+<!-- moseq_train.FullFitTask&#45;&gt;moseq_train.FullFit -->
+<g id="edge10" class="edge">
+<title>moseq_train.FullFitTask-&gt;moseq_train.FullFit</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-61.66C319.84,-50.36 319.84,-35.66 319.84,-25.58"/>
 </g>
-<!-- moseq_train.Bodyparts -->
+<!-- moseq_infer.PoseEstimationMethod -->
 <g id="node12" class="node">
-<title>moseq_train.Bodyparts</title>
-<g id="a_node12"><a xlink:title="→ moseq_train.KeypointSet
bodyparts_id         
------------------------------
anterior_bodyparts   
posterior_bodyparts  
use_bodyparts        
bodyparts_desc=&quot;&quot;    
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="377.34,-370.08 238.34,-370.08 238.34,-335.52 377.34,-335.52 377.34,-370.08"/>
-<text text-anchor="start" x="246.34" y="-349.53" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.Bodyparts</text>
+<title>moseq_infer.PoseEstimationMethod</title>
+<g id="a_node12"><a xlink:title="pose_estimation_method 
------------------------------
pose_estimation_desc 
">
+<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="583.59,-511.2 410.09,-511.2 410.09,-476.64 583.59,-476.64 583.59,-511.2"/>
+<text text-anchor="start" x="418.09" y="-491.42" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.PoseEstimationMethod</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.Bodyparts -->
-<g id="edge10" class="edge">
-<title>moseq_train.KeypointSet-&gt;moseq_train.Bodyparts</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M437.77,-405.59C411.01,-394.92 376.51,-381.17 349.78,-370.52"/>
+<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_train.KeypointSet -->
+<g id="edge11" class="edge">
+<title>moseq_infer.PoseEstimationMethod-&gt;moseq_train.KeypointSet</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M495.66,-476.67C494.87,-465.9 493.84,-451.81 493.05,-440.97"/>
 </g>
-<!-- moseq_train.KeypointSet.VideoFile -->
-<g id="node19" class="node">
-<title>moseq_train.KeypointSet.VideoFile</title>
-<g id="a_node19"><a xlink:title="→ moseq_train.KeypointSet
video_id             
------------------------------
video_path           
">
-<polygon fill="none" stroke="none" points="564.34,-362.8 395.34,-362.8 395.34,-342.8 564.34,-342.8 564.34,-362.8"/>
-<text text-anchor="start" x="403.34" y="-350.3" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.KeypointSet.VideoFile</text>
+<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_infer.InferenceTask -->
+<g id="edge12" class="edge">
+<title>moseq_infer.PoseEstimationMethod-&gt;moseq_infer.InferenceTask</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M536.42,-476.67C562.56,-465.94 596.72,-451.91 623.1,-441.09"/>
+</g>
+<!-- moseq_infer.Model -->
+<g id="node14" class="node">
+<title>moseq_infer.Model</title>
+<g id="a_node14"><a xlink:title="model_id             
------------------------------
model_name           
model_dir            
latent_dim           
kappa                
model_desc=&quot;&quot;        
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="722.47,-511.2 605.22,-511.2 605.22,-476.64 722.47,-476.64 722.47,-511.2"/>
+<text text-anchor="start" x="613.22" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.Model</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.KeypointSet.VideoFile -->
-<g id="edge11" class="edge">
-<title>moseq_train.KeypointSet-&gt;moseq_train.KeypointSet.VideoFile</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M479.84,-405.76C479.84,-392.13 479.84,-373.36 479.84,-362.51"/>
+<!-- moseq_infer.Model&#45;&gt;moseq_infer.InferenceTask -->
+<g id="edge13" class="edge">
+<title>moseq_infer.Model-&gt;moseq_infer.InferenceTask</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M663.84,-476.32C663.84,-465.54 663.84,-451.55 663.84,-440.8"/>
 </g>
-<!-- moseq_train.PreFit -->
-<g id="node9" class="node">
-<title>moseq_train.PreFit</title>
-<g id="a_node9"><a xlink:title="→ moseq_train.PreFitTask
------------------------------
model_name=&quot;&quot;        
pre_fit_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="445.84" cy="-12.96" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="445.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PreFit</text>
+<!-- moseq_infer.Inference.MotionSequence -->
+<g id="node15" class="node">
+<title>moseq_infer.Inference.MotionSequence</title>
+<g id="a_node15"><a xlink:title="→ moseq_infer.Inference
video_name           
------------------------------
syllable             
latent_state         
centroid             
heading              
">
+<polygon fill="none" stroke="none" points="925.09,-292.24 736.59,-292.24 736.59,-272.24 925.09,-272.24 925.09,-292.24"/>
+<text text-anchor="start" x="744.59" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.MotionSequence</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.FullFitTask&#45;&gt;moseq_train.FullFit -->
-<g id="edge12" class="edge">
-<title>moseq_train.FullFitTask-&gt;moseq_train.FullFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.84,-61.99C307.84,-50.69 307.84,-35.88 307.84,-25.71"/>
-</g>
 <!-- moseq_train.PreFitTask&#45;&gt;moseq_train.PreFit -->
-<g id="edge13" class="edge">
-<title>moseq_train.PreFitTask-&gt;moseq_train.PreFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M445.84,-61.99C445.84,-50.69 445.84,-35.88 445.84,-25.71"/>
-</g>
-<!-- moseq_train.Bodyparts&#45;&gt;moseq_train.PCATask -->
 <g id="edge14" class="edge">
-<title>moseq_train.Bodyparts-&gt;moseq_train.PCATask</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.84,-335.2C307.84,-324.42 307.84,-310.43 307.84,-299.68"/>
+<title>moseq_train.PreFitTask-&gt;moseq_train.PreFit</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M480.84,-61.66C480.84,-50.36 480.84,-35.66 480.84,-25.58"/>
 </g>
-<!-- moseq_train.PCAPrep&#45;&gt;moseq_train.PCAFit -->
+<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.GridMoviesSampledInstances -->
 <g id="edge15" class="edge">
-<title>moseq_train.PCAPrep-&gt;moseq_train.PCAFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.84,-194.14C307.84,-182.84 307.84,-168.14 307.84,-158.06"/>
+<title>moseq_infer.Inference-&gt;moseq_infer.Inference.GridMoviesSampledInstances</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M655.23,-343.12C642.22,-330 617.69,-305.27 604.6,-292.07"/>
 </g>
-<!-- moseq_infer.InferenceTask&#45;&gt;moseq_infer.Inference -->
+<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.MotionSequence -->
 <g id="edge16" class="edge">
-<title>moseq_infer.InferenceTask-&gt;moseq_infer.Inference</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M651.84,-405.76C651.84,-393.4 651.84,-376.81 651.84,-365.73"/>
+<title>moseq_infer.Inference-&gt;moseq_infer.Inference.MotionSequence</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M675.32,-347.09C702.88,-335.77 773.16,-306.92 809.04,-292.19"/>
 </g>
-<!-- moseq_infer.PoseEstimationMethod -->
-<g id="node17" class="node">
-<title>moseq_infer.PoseEstimationMethod</title>
-<g id="a_node17"><a xlink:title="pose_estimation_method 
------------------------------
pose_estimation_desc 
">
-<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="571.59,-511.2 398.09,-511.2 398.09,-476.64 571.59,-476.64 571.59,-511.2"/>
-<text text-anchor="start" x="406.09" y="-491.42" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.PoseEstimationMethod</text>
+<!-- moseq_infer.VideoRecording -->
+<g id="node19" class="node">
+<title>moseq_infer.VideoRecording</title>
+<g id="a_node19"><a xlink:title="→ session.Session
recording_id         
------------------------------
→ lab.Device
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="926.09,-511.2 755.59,-511.2 755.59,-476.64 926.09,-476.64 926.09,-511.2"/>
+<text text-anchor="start" x="763.59" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.VideoRecording</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_train.KeypointSet -->
+<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.VideoRecording.File -->
 <g id="edge17" class="edge">
-<title>moseq_infer.PoseEstimationMethod-&gt;moseq_train.KeypointSet</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M483.66,-476.67C482.87,-465.9 481.84,-451.81 481.05,-440.97"/>
+<title>moseq_infer.VideoRecording-&gt;moseq_infer.VideoRecording.File</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M840.84,-476.32C840.84,-462.69 840.84,-443.92 840.84,-433.07"/>
 </g>
-<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_infer.InferenceTask -->
+<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.InferenceTask -->
 <g id="edge18" class="edge">
-<title>moseq_infer.PoseEstimationMethod-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M524.42,-476.67C550.56,-465.94 584.72,-451.91 611.1,-441.09"/>
+<title>moseq_infer.VideoRecording-&gt;moseq_infer.InferenceTask</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M797.54,-476.15C770.01,-465.48 734.5,-451.73 707,-441.08"/>
 </g>
 </g>
 </svg>
\ No newline at end of file
diff --git a/images/pipeline_moseq_infer.svg b/images/pipeline_moseq_infer.svg
index cfc9612..b3a8da2 100644
--- a/images/pipeline_moseq_infer.svg
+++ b/images/pipeline_moseq_infer.svg
@@ -1,112 +1,112 @@
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="505pt" height="231pt" viewBox="0.00 0.00 504.88 231.04">
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="512pt" height="231pt" viewBox="0.00 0.00 512.00 231.04">
 <g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 227.04)">
-<polygon fill="white" stroke="none" points="-4,4 -4,-227.04 500.88,-227.04 500.88,4 -4,4"/>
-<!-- moseq_infer.VideoRecording.File -->
-<g id="node1" class="node">
-<title>moseq_infer.VideoRecording.File</title>
-<g id="a_node1"><a xlink:title="→ moseq_infer.VideoRecording
file_id              
------------------------------
file_path            
">
-<polygon fill="none" stroke="none" points="165.25,-145.2 5.25,-145.2 5.25,-125.2 165.25,-125.2 165.25,-145.2"/>
-<text text-anchor="start" x="13.25" y="-132.7" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.VideoRecording.File</text>
-</a>
-</g>
-</g>
+<polygon fill="white" stroke="none" points="-4,4 -4,-227.04 508,-227.04 508,4 -4,4"/>
 <!-- moseq_infer.Inference.GridMoviesSampledInstances -->
-<g id="node2" class="node">
+<g id="node1" class="node">
 <title>moseq_infer.Inference.GridMoviesSampledInstances</title>
-<g id="a_node2"><a xlink:title="→ moseq_infer.Inference
syllable             
------------------------------
instances            
">
-<polygon fill="none" stroke="none" points="281.38,-20 35.12,-20 35.12,0 281.38,0 281.38,-20"/>
-<text text-anchor="start" x="43.12" y="-7.5" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.GridMoviesSampledInstances</text>
-</a>
-</g>
-</g>
-<!-- moseq_infer.Inference.MotionSequence -->
-<g id="node3" class="node">
-<title>moseq_infer.Inference.MotionSequence</title>
-<g id="a_node3"><a xlink:title="→ moseq_infer.Inference
video_name           
------------------------------
syllable             
latent_state         
centroid             
heading              
">
-<polygon fill="none" stroke="none" points="487.5,-20 299,-20 299,0 487.5,0 487.5,-20"/>
-<text text-anchor="start" x="307" y="-7.5" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.MotionSequence</text>
+<g id="a_node1"><a xlink:title="→ moseq_infer.Inference
syllable             
------------------------------
instances            
">
+<polygon fill="none" stroke="none" points="247.88,-20 1.62,-20 1.62,0 247.88,0 247.88,-20"/>
+<text text-anchor="start" x="9.62" y="-7.5" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.GridMoviesSampledInstances</text>
 </a>
 </g>
 </g>
 <!-- moseq_infer.Inference -->
-<g id="node4" class="node">
+<g id="node2" class="node">
 <title>moseq_infer.Inference</title>
-<g id="a_node4"><a xlink:title="→ moseq_infer.InferenceTask
------------------------------
inference_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="275.25" cy="-68.96" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="275.25" y="-64.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_infer.Inference</text>
+<g id="a_node2"><a xlink:title="→ moseq_infer.InferenceTask
------------------------------
inference_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="241.75" cy="-68.96" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="241.75" y="-64.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_infer.Inference</text>
 </a>
 </g>
 </g>
 <!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.GridMoviesSampledInstances -->
 <g id="edge1" class="edge">
 <title>moseq_infer.Inference-&gt;moseq_infer.Inference.GridMoviesSampledInstances</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M264.03,-62.5C243.78,-52.64 200.97,-31.8 176.47,-19.87"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M230.53,-62.5C210.28,-52.64 167.47,-31.8 142.97,-19.87"/>
+</g>
+<!-- moseq_infer.Inference.MotionSequence -->
+<g id="node7" class="node">
+<title>moseq_infer.Inference.MotionSequence</title>
+<g id="a_node7"><a xlink:title="→ moseq_infer.Inference
video_name           
------------------------------
syllable             
latent_state         
centroid             
heading              
">
+<polygon fill="none" stroke="none" points="454,-20 265.5,-20 265.5,0 454,0 454,-20"/>
+<text text-anchor="start" x="273.5" y="-7.5" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.MotionSequence</text>
+</a>
+</g>
 </g>
 <!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.MotionSequence -->
 <g id="edge2" class="edge">
 <title>moseq_infer.Inference-&gt;moseq_infer.Inference.MotionSequence</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M286.14,-62.7C306.34,-52.95 349.81,-31.97 374.72,-19.95"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M252.64,-62.7C272.84,-52.95 316.31,-31.97 341.22,-19.95"/>
+</g>
+<!-- moseq_infer.PoseEstimationMethod -->
+<g id="node3" class="node">
+<title>moseq_infer.PoseEstimationMethod</title>
+<g id="a_node3"><a xlink:title="pose_estimation_method 
------------------------------
pose_estimation_desc 
">
+<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="173.5,-223.04 0,-223.04 0,-188.48 173.5,-188.48 173.5,-223.04"/>
+<text text-anchor="start" x="8" y="-203.26" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.PoseEstimationMethod</text>
+</a>
+</g>
 </g>
 <!-- moseq_infer.InferenceTask -->
 <g id="node5" class="node">
 <title>moseq_infer.InferenceTask</title>
 <g id="a_node5"><a xlink:title="→ moseq_infer.VideoRecording
→ moseq_infer.Model
------------------------------
→ moseq_infer.PoseEstimationMethod
keypointset_dir      
inference_output_dir=&quot;&quot; 
inference_desc=&quot;&quot;    
num_iterations=null  
task_mode=&quot;load&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="354.5,-152.48 196,-152.48 196,-117.92 354.5,-117.92 354.5,-152.48"/>
-<text text-anchor="middle" x="275.25" y="-130.93" font-family="arial" font-size="12.00" fill="darkgreen">moseq_infer.InferenceTask</text>
-</a>
-</g>
-</g>
-<!-- moseq_infer.InferenceTask&#45;&gt;moseq_infer.Inference -->
-<g id="edge3" class="edge">
-<title>moseq_infer.InferenceTask-&gt;moseq_infer.Inference</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M275.25,-117.66C275.25,-106.36 275.25,-91.66 275.25,-81.58"/>
-</g>
-<!-- moseq_infer.PoseEstimationMethod -->
-<g id="node6" class="node">
-<title>moseq_infer.PoseEstimationMethod</title>
-<g id="a_node6"><a xlink:title="pose_estimation_method 
------------------------------
pose_estimation_desc 
">
-<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="362,-223.04 188.5,-223.04 188.5,-188.48 362,-188.48 362,-223.04"/>
-<text text-anchor="start" x="196.5" y="-203.26" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.PoseEstimationMethod</text>
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="321,-152.48 162.5,-152.48 162.5,-117.92 321,-117.92 321,-152.48"/>
+<text text-anchor="middle" x="241.75" y="-130.93" font-family="arial" font-size="12.00" fill="darkgreen">moseq_infer.InferenceTask</text>
 </a>
 </g>
 </g>
 <!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge4" class="edge">
+<g id="edge3" class="edge">
 <title>moseq_infer.PoseEstimationMethod-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M275.25,-188.51C275.25,-177.74 275.25,-163.65 275.25,-152.81"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M123.48,-188.51C147.74,-177.78 179.46,-163.75 203.93,-152.93"/>
 </g>
 <!-- moseq_infer.Model -->
-<g id="node7" class="node">
+<g id="node4" class="node">
 <title>moseq_infer.Model</title>
-<g id="a_node7"><a xlink:title="model_id             
------------------------------
model_name           
model_dir            
latent_dim           
kappa                
model_desc=&quot;&quot;        
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="496.88,-223.04 379.62,-223.04 379.62,-188.48 496.88,-188.48 496.88,-223.04"/>
-<text text-anchor="start" x="387.62" y="-202.49" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.Model</text>
+<g id="a_node4"><a xlink:title="model_id             
------------------------------
model_name           
model_dir            
latent_dim           
kappa                
model_desc=&quot;&quot;        
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="308.38,-223.04 191.12,-223.04 191.12,-188.48 308.38,-188.48 308.38,-223.04"/>
+<text text-anchor="start" x="199.12" y="-202.49" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.Model</text>
 </a>
 </g>
 </g>
 <!-- moseq_infer.Model&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge5" class="edge">
+<g id="edge4" class="edge">
 <title>moseq_infer.Model-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M398.38,-187.99C373.02,-177.32 340.32,-163.57 314.99,-152.92"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M247.81,-188.16C246.56,-177.38 244.92,-163.39 243.67,-152.64"/>
+</g>
+<!-- moseq_infer.InferenceTask&#45;&gt;moseq_infer.Inference -->
+<g id="edge5" class="edge">
+<title>moseq_infer.InferenceTask-&gt;moseq_infer.Inference</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M241.75,-117.66C241.75,-106.36 241.75,-91.66 241.75,-81.58"/>
+</g>
+<!-- moseq_infer.VideoRecording.File -->
+<g id="node6" class="node">
+<title>moseq_infer.VideoRecording.File</title>
+<g id="a_node6"><a xlink:title="→ moseq_infer.VideoRecording
file_id              
------------------------------
file_path            
">
+<polygon fill="none" stroke="none" points="498.75,-145.2 338.75,-145.2 338.75,-125.2 498.75,-125.2 498.75,-145.2"/>
+<text text-anchor="start" x="346.75" y="-132.7" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.VideoRecording.File</text>
+</a>
+</g>
 </g>
 <!-- moseq_infer.VideoRecording -->
 <g id="node8" class="node">
 <title>moseq_infer.VideoRecording</title>
 <g id="a_node8"><a xlink:title="→ session.Session
recording_id         
------------------------------
→ lab.Device
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="170.5,-223.04 0,-223.04 0,-188.48 170.5,-188.48 170.5,-223.04"/>
-<text text-anchor="start" x="8" y="-202.49" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.VideoRecording</text>
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="504,-223.04 333.5,-223.04 333.5,-188.48 504,-188.48 504,-223.04"/>
+<text text-anchor="start" x="341.5" y="-202.49" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.VideoRecording</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.VideoRecording.File -->
+<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.InferenceTask -->
 <g id="edge6" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.VideoRecording.File</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M85.25,-188.16C85.25,-174.53 85.25,-155.76 85.25,-144.91"/>
+<title>moseq_infer.VideoRecording-&gt;moseq_infer.InferenceTask</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M375.45,-187.99C347.92,-177.32 312.41,-163.57 284.91,-152.92"/>
 </g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.InferenceTask -->
+<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.VideoRecording.File -->
 <g id="edge7" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M131.73,-187.99C161.28,-177.32 199.4,-163.57 228.92,-152.92"/>
+<title>moseq_infer.VideoRecording-&gt;moseq_infer.VideoRecording.File</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M418.75,-188.16C418.75,-174.53 418.75,-155.76 418.75,-144.91"/>
 </g>
 </g>
 </svg>
\ No newline at end of file
diff --git a/images/pipeline_moseq_train.svg b/images/pipeline_moseq_train.svg
index a1978fe..be90571 100644
--- a/images/pipeline_moseq_train.svg
+++ b/images/pipeline_moseq_train.svg
@@ -1,266 +1,154 @@
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="934pt" height="519pt" viewBox="0.00 0.00 934.09 519.20">
-<g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 515.2)">
-<polygon fill="white" stroke="none" points="-4,4 -4,-515.2 930.09,-515.2 930.09,4 -4,4"/>
-<!-- moseq_infer.Inference.GridMoviesSampledInstances -->
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="560pt" height="449pt" viewBox="0.00 0.00 559.75 448.64">
+<g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 444.64)">
+<polygon fill="white" stroke="none" points="-4,4 -4,-444.64 555.75,-444.64 555.75,4 -4,4"/>
+<!-- moseq_train.KeypointSet -->
 <g id="node1" class="node">
-<title>moseq_infer.Inference.GridMoviesSampledInstances</title>
-<g id="a_node1"><a xlink:title="→ moseq_infer.Inference
syllable             
------------------------------
instances            
">
-<polygon fill="none" stroke="none" points="718.97,-292.24 472.72,-292.24 472.72,-272.24 718.97,-272.24 718.97,-292.24"/>
-<text text-anchor="start" x="480.72" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.GridMoviesSampledInstances</text>
+<title>moseq_train.KeypointSet</title>
+<g id="a_node1"><a xlink:title="kpset_id             
------------------------------
→ moseq_infer.PoseEstimationMethod
kpset_dir            
kpset_desc=&quot;&quot;        
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="179.88,-440.64 29.62,-440.64 29.62,-406.08 179.88,-406.08 179.88,-440.64"/>
+<text text-anchor="start" x="37.62" y="-420.09" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.KeypointSet</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.PreFit -->
+<!-- moseq_train.KeypointSet.VideoFile -->
 <g id="node2" class="node">
-<title>moseq_train.PreFit</title>
-<g id="a_node2"><a xlink:title="→ moseq_train.PreFitTask
------------------------------
model_name=&quot;&quot;        
pre_fit_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="480.84" cy="-12.96" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="480.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PreFit</text>
-</a>
-</g>
-</g>
-<!-- moseq_train.PCAPrep -->
-<g id="node3" class="node">
-<title>moseq_train.PCAPrep</title>
-<g id="a_node3"><a xlink:title="→ moseq_train.PCATask
------------------------------
coordinates          
confidences          
formatted_bodyparts  
average_frame_rate   
frame_rates          
">
-<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="319.84" cy="-211.68" rx="88.87" ry="17.28"/>
-<text text-anchor="middle" x="319.84" y="-207.41" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.PCAPrep</text>
-</a>
-</g>
-</g>
-<!-- moseq_train.PCAFit -->
-<g id="node7" class="node">
-<title>moseq_train.PCAFit</title>
-<g id="a_node7"><a xlink:title="→ moseq_train.PCAPrep
------------------------------
pca_fit_time=null    
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="319.84" cy="-145.44" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="319.84" y="-141.16" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PCAFit</text>
+<title>moseq_train.KeypointSet.VideoFile</title>
+<g id="a_node2"><a xlink:title="→ moseq_train.KeypointSet
video_id             
------------------------------
video_path           
">
+<polygon fill="none" stroke="none" points="189.25,-362.8 20.25,-362.8 20.25,-342.8 189.25,-342.8 189.25,-362.8"/>
+<text text-anchor="start" x="28.25" y="-350.3" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.KeypointSet.VideoFile</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.PCAPrep&#45;&gt;moseq_train.PCAFit -->
+<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.KeypointSet.VideoFile -->
 <g id="edge1" class="edge">
-<title>moseq_train.PCAPrep-&gt;moseq_train.PCAFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-194.14C319.84,-182.84 319.84,-168.14 319.84,-158.06"/>
-</g>
-<!-- moseq_train.LatentDimension -->
-<g id="node4" class="node">
-<title>moseq_train.LatentDimension</title>
-<g id="a_node4"><a xlink:title="→ moseq_train.PCAFit
------------------------------
variance_percentage  
latent_dimension     
latent_dim_desc      
">
-<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="114.84" cy="-79.2" rx="114.84" ry="17.28"/>
-<text text-anchor="middle" x="114.84" y="-74.92" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.LatentDimension</text>
-</a>
-</g>
-</g>
-<!-- moseq_infer.VideoRecording.File -->
-<g id="node5" class="node">
-<title>moseq_infer.VideoRecording.File</title>
-<g id="a_node5"><a xlink:title="→ moseq_infer.VideoRecording
file_id              
------------------------------
file_path            
">
-<polygon fill="none" stroke="none" points="920.84,-433.36 760.84,-433.36 760.84,-413.36 920.84,-413.36 920.84,-433.36"/>
-<text text-anchor="start" x="768.84" y="-420.86" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.VideoRecording.File</text>
-</a>
-</g>
-</g>
-<!-- moseq_train.KeypointSet -->
-<g id="node6" class="node">
-<title>moseq_train.KeypointSet</title>
-<g id="a_node6"><a xlink:title="kpset_id             
------------------------------
→ moseq_infer.PoseEstimationMethod
kpset_dir            
kpset_desc=&quot;&quot;        
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="566.97,-440.64 416.72,-440.64 416.72,-406.08 566.97,-406.08 566.97,-440.64"/>
-<text text-anchor="start" x="424.72" y="-420.09" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.KeypointSet</text>
-</a>
-</g>
+<title>moseq_train.KeypointSet-&gt;moseq_train.KeypointSet.VideoFile</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M104.75,-405.76C104.75,-392.13 104.75,-373.36 104.75,-362.51"/>
 </g>
 <!-- moseq_train.Bodyparts -->
-<g id="node10" class="node">
+<g id="node11" class="node">
 <title>moseq_train.Bodyparts</title>
-<g id="a_node10"><a xlink:title="→ moseq_train.KeypointSet
bodyparts_id         
------------------------------
anterior_bodyparts   
posterior_bodyparts  
use_bodyparts        
bodyparts_desc=&quot;&quot;    
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="389.34,-370.08 250.34,-370.08 250.34,-335.52 389.34,-335.52 389.34,-370.08"/>
-<text text-anchor="start" x="258.34" y="-349.53" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.Bodyparts</text>
+<g id="a_node11"><a xlink:title="→ moseq_train.KeypointSet
bodyparts_id         
------------------------------
anterior_bodyparts   
posterior_bodyparts  
use_bodyparts        
bodyparts_desc=&quot;&quot;    
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="346.25,-370.08 207.25,-370.08 207.25,-335.52 346.25,-335.52 346.25,-370.08"/>
+<text text-anchor="start" x="215.25" y="-349.53" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.Bodyparts</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.KeypointSet&#45;&gt;moseq_train.Bodyparts -->
 <g id="edge2" class="edge">
 <title>moseq_train.KeypointSet-&gt;moseq_train.Bodyparts</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M449.77,-405.59C423.01,-394.92 388.51,-381.17 361.78,-370.52"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M146.83,-405.59C173.58,-394.92 208.08,-381.17 234.81,-370.52"/>
 </g>
-<!-- moseq_train.KeypointSet.VideoFile -->
-<g id="node17" class="node">
-<title>moseq_train.KeypointSet.VideoFile</title>
-<g id="a_node17"><a xlink:title="→ moseq_train.KeypointSet
video_id             
------------------------------
video_path           
">
-<polygon fill="none" stroke="none" points="576.34,-362.8 407.34,-362.8 407.34,-342.8 576.34,-342.8 576.34,-362.8"/>
-<text text-anchor="start" x="415.34" y="-350.3" font-family="arial" text-decoration="underline" font-size="10.00">moseq_train.KeypointSet.VideoFile</text>
+<!-- moseq_train.PreFit -->
+<g id="node3" class="node">
+<title>moseq_train.PreFit</title>
+<g id="a_node3"><a xlink:title="→ moseq_train.PreFitTask
------------------------------
model_name=&quot;&quot;        
pre_fit_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="480.75" cy="-12.96" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="480.75" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PreFit</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.KeypointSet&#45;&gt;moseq_train.KeypointSet.VideoFile -->
-<g id="edge3" class="edge">
-<title>moseq_train.KeypointSet-&gt;moseq_train.KeypointSet.VideoFile</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M491.84,-405.76C491.84,-392.13 491.84,-373.36 491.84,-362.51"/>
+<!-- moseq_train.PCAFit -->
+<g id="node4" class="node">
+<title>moseq_train.PCAFit</title>
+<g id="a_node4"><a xlink:title="→ moseq_train.PCAPrep
------------------------------
pca_fit_time=null    
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="276.75" cy="-145.44" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="276.75" y="-141.16" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.PCAFit</text>
+</a>
 </g>
-<!-- moseq_train.PCAFit&#45;&gt;moseq_train.LatentDimension -->
-<g id="edge4" class="edge">
-<title>moseq_train.PCAFit-&gt;moseq_train.LatentDimension</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M307.93,-140.71C280.33,-132.06 210.71,-110.24 163.07,-95.31"/>
 </g>
 <!-- moseq_train.FullFitTask -->
-<g id="node11" class="node">
+<g id="node5" class="node">
 <title>moseq_train.FullFitTask</title>
-<g id="a_node11"><a xlink:title="→ moseq_train.PCAFit
full_latent_dim      
full_kappa           
full_num_iterations  
------------------------------
model_name           
task_mode=&quot;load&quot;     
full_fit_desc=&quot;&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="391.59,-96.48 248.09,-96.48 248.09,-61.92 391.59,-61.92 391.59,-96.48"/>
-<text text-anchor="start" x="256.09" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.FullFitTask</text>
+<g id="a_node5"><a xlink:title="→ moseq_train.PCAFit
full_latent_dim      
full_kappa           
full_num_iterations  
------------------------------
model_name           
task_mode=&quot;load&quot;     
full_fit_desc=&quot;&quot;     
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="143.5,-96.48 0,-96.48 0,-61.92 143.5,-61.92 143.5,-96.48"/>
+<text text-anchor="start" x="8" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.FullFitTask</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.PCAFit&#45;&gt;moseq_train.FullFitTask -->
-<g id="edge5" class="edge">
+<g id="edge3" class="edge">
 <title>moseq_train.PCAFit-&gt;moseq_train.FullFitTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M319.84,-132.66C319.84,-122.52 319.84,-107.78 319.84,-96.5"/>
-</g>
-<!-- moseq_train.PreFitTask -->
-<g id="node16" class="node">
-<title>moseq_train.PreFitTask</title>
-<g id="a_node16"><a xlink:title="→ moseq_train.PCAFit
pre_latent_dim       
pre_kappa            
pre_num_iterations   
------------------------------
model_name           
task_mode=&quot;load&quot;     
pre_fit_desc=&quot;&quot;      
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="551.84,-96.48 409.84,-96.48 409.84,-61.92 551.84,-61.92 551.84,-96.48"/>
-<text text-anchor="start" x="417.84" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.PreFitTask</text>
-</a>
-</g>
-</g>
-<!-- moseq_train.PCAFit&#45;&gt;moseq_train.PreFitTask -->
-<g id="edge6" class="edge">
-<title>moseq_train.PCAFit-&gt;moseq_train.PreFitTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M331.42,-139.82C353.55,-130.99 402.87,-111.31 438.81,-96.97"/>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M264.84,-140.71C238.18,-132.35 172.31,-111.71 124.92,-96.86"/>
 </g>
-<!-- moseq_infer.InferenceTask -->
-<g id="node8" class="node">
-<title>moseq_infer.InferenceTask</title>
-<g id="a_node8"><a xlink:title="→ moseq_infer.VideoRecording
→ moseq_infer.Model
------------------------------
→ moseq_infer.PoseEstimationMethod
keypointset_dir      
inference_output_dir=&quot;&quot; 
inference_desc=&quot;&quot;    
num_iterations=null  
task_mode=&quot;load&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="743.09,-440.64 584.59,-440.64 584.59,-406.08 743.09,-406.08 743.09,-440.64"/>
-<text text-anchor="middle" x="663.84" y="-419.09" font-family="arial" font-size="12.00" fill="darkgreen">moseq_infer.InferenceTask</text>
-</a>
-</g>
-</g>
-<!-- moseq_infer.Inference -->
-<g id="node18" class="node">
-<title>moseq_infer.Inference</title>
-<g id="a_node18"><a xlink:title="→ moseq_infer.InferenceTask
------------------------------
inference_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="663.84" cy="-352.8" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="663.84" y="-348.53" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_infer.Inference</text>
+<!-- moseq_train.LatentDimension -->
+<g id="node7" class="node">
+<title>moseq_train.LatentDimension</title>
+<g id="a_node7"><a xlink:title="→ moseq_train.PCAFit
------------------------------
variance_percentage  
latent_dimension     
latent_dim_desc      
">
+<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="276.75" cy="-79.2" rx="114.84" ry="17.28"/>
+<text text-anchor="middle" x="276.75" y="-74.92" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.LatentDimension</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.InferenceTask&#45;&gt;moseq_infer.Inference -->
-<g id="edge7" class="edge">
-<title>moseq_infer.InferenceTask-&gt;moseq_infer.Inference</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M663.84,-405.76C663.84,-393.4 663.84,-376.81 663.84,-365.73"/>
+<!-- moseq_train.PCAFit&#45;&gt;moseq_train.LatentDimension -->
+<g id="edge4" class="edge">
+<title>moseq_train.PCAFit-&gt;moseq_train.LatentDimension</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M276.75,-132.66C276.75,-122.52 276.75,-107.78 276.75,-96.5"/>
 </g>
-<!-- moseq_train.PCATask -->
+<!-- moseq_train.PreFitTask -->
 <g id="node9" class="node">
-<title>moseq_train.PCATask</title>
-<g id="a_node9"><a xlink:title="→ moseq_train.Bodyparts
------------------------------
kpms_project_output_dir=&quot;&quot; 
task_mode=&quot;load&quot;     
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="386.72,-299.52 252.97,-299.52 252.97,-264.96 386.72,-264.96 386.72,-299.52"/>
-<text text-anchor="middle" x="319.84" y="-277.97" font-family="arial" font-size="12.00" fill="darkgreen">moseq_train.PCATask</text>
+<title>moseq_train.PreFitTask</title>
+<g id="a_node9"><a xlink:title="→ moseq_train.PCAFit
pre_latent_dim       
pre_kappa            
pre_num_iterations   
------------------------------
model_name           
task_mode=&quot;load&quot;     
pre_fit_desc=&quot;&quot;      
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="551.75,-96.48 409.75,-96.48 409.75,-61.92 551.75,-61.92 551.75,-96.48"/>
+<text text-anchor="start" x="417.75" y="-75.92" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_train.PreFitTask</text>
 </a>
 </g>
 </g>
-<!-- moseq_train.PCATask&#45;&gt;moseq_train.PCAPrep -->
-<g id="edge8" class="edge">
-<title>moseq_train.PCATask-&gt;moseq_train.PCAPrep</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-264.64C319.84,-253.86 319.84,-239.87 319.84,-229.12"/>
-</g>
-<!-- moseq_train.Bodyparts&#45;&gt;moseq_train.PCATask -->
-<g id="edge9" class="edge">
-<title>moseq_train.Bodyparts-&gt;moseq_train.PCATask</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-335.2C319.84,-324.42 319.84,-310.43 319.84,-299.68"/>
+<!-- moseq_train.PCAFit&#45;&gt;moseq_train.PreFitTask -->
+<g id="edge5" class="edge">
+<title>moseq_train.PCAFit-&gt;moseq_train.PreFitTask</title>
+<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M288.6,-140.71C315.13,-132.35 380.68,-111.71 427.84,-96.86"/>
 </g>
 <!-- moseq_train.FullFit -->
-<g id="node13" class="node">
+<g id="node8" class="node">
 <title>moseq_train.FullFit</title>
-<g id="a_node13"><a xlink:title="→ moseq_train.FullFitTask
------------------------------
model_name           
full_fit_duration=null 
">
-<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="319.84" cy="-12.96" rx="12.96" ry="12.96"/>
-<text text-anchor="middle" x="319.84" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.FullFit</text>
+<g id="a_node8"><a xlink:title="→ moseq_train.FullFitTask
------------------------------
model_name           
full_fit_duration=null 
">
+<ellipse fill="#ff0000" fill-opacity="0.125490" stroke="#ff0000" stroke-opacity="0.125490" cx="71.75" cy="-12.96" rx="12.96" ry="12.96"/>
+<text text-anchor="middle" x="71.75" y="-8.69" font-family="arial" font-size="12.00" fill="#7f0000" fill-opacity="0.627451">moseq_train.FullFit</text>
 </a>
 </g>
 </g>
 <!-- moseq_train.FullFitTask&#45;&gt;moseq_train.FullFit -->
-<g id="edge10" class="edge">
+<g id="edge6" class="edge">
 <title>moseq_train.FullFitTask-&gt;moseq_train.FullFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M319.84,-61.66C319.84,-50.36 319.84,-35.66 319.84,-25.58"/>
-</g>
-<!-- moseq_infer.PoseEstimationMethod -->
-<g id="node12" class="node">
-<title>moseq_infer.PoseEstimationMethod</title>
-<g id="a_node12"><a xlink:title="pose_estimation_method 
------------------------------
pose_estimation_desc 
">
-<polygon fill="#000000" fill-opacity="0.125490" stroke="none" points="583.59,-511.2 410.09,-511.2 410.09,-476.64 583.59,-476.64 583.59,-511.2"/>
-<text text-anchor="start" x="418.09" y="-491.42" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.PoseEstimationMethod</text>
-</a>
-</g>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M71.75,-61.66C71.75,-50.36 71.75,-35.66 71.75,-25.58"/>
 </g>
-<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_train.KeypointSet -->
-<g id="edge11" class="edge">
-<title>moseq_infer.PoseEstimationMethod-&gt;moseq_train.KeypointSet</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M495.66,-476.67C494.87,-465.9 493.84,-451.81 493.05,-440.97"/>
-</g>
-<!-- moseq_infer.PoseEstimationMethod&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge12" class="edge">
-<title>moseq_infer.PoseEstimationMethod-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-dasharray="5,2" stroke-opacity="0.250980" d="M536.42,-476.67C562.56,-465.94 596.72,-451.91 623.1,-441.09"/>
-</g>
-<!-- moseq_infer.Model -->
-<g id="node14" class="node">
-<title>moseq_infer.Model</title>
-<g id="a_node14"><a xlink:title="model_id             
------------------------------
model_name           
model_dir            
latent_dim           
kappa                
model_desc=&quot;&quot;        
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="722.47,-511.2 605.22,-511.2 605.22,-476.64 722.47,-476.64 722.47,-511.2"/>
-<text text-anchor="start" x="613.22" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.Model</text>
+<!-- moseq_train.PCAPrep -->
+<g id="node6" class="node">
+<title>moseq_train.PCAPrep</title>
+<g id="a_node6"><a xlink:title="→ moseq_train.PCATask
------------------------------
coordinates          
confidences          
formatted_bodyparts  
average_frame_rate   
frame_rates          
">
+<ellipse fill="#00007f" fill-opacity="0.250980" stroke="#00007f" stroke-opacity="0.250980" cx="276.75" cy="-211.68" rx="88.87" ry="17.28"/>
+<text text-anchor="middle" x="276.75" y="-207.41" font-family="arial" font-size="12.00" fill="#00007f" fill-opacity="0.627451">moseq_train.PCAPrep</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.Model&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge13" class="edge">
-<title>moseq_infer.Model-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M663.84,-476.32C663.84,-465.54 663.84,-451.55 663.84,-440.8"/>
-</g>
-<!-- moseq_infer.Inference.MotionSequence -->
-<g id="node15" class="node">
-<title>moseq_infer.Inference.MotionSequence</title>
-<g id="a_node15"><a xlink:title="→ moseq_infer.Inference
video_name           
------------------------------
syllable             
latent_state         
centroid             
heading              
">
-<polygon fill="none" stroke="none" points="925.09,-292.24 736.59,-292.24 736.59,-272.24 925.09,-272.24 925.09,-292.24"/>
-<text text-anchor="start" x="744.59" y="-279.74" font-family="arial" text-decoration="underline" font-size="10.00">moseq_infer.Inference.MotionSequence</text>
-</a>
-</g>
+<!-- moseq_train.PCAPrep&#45;&gt;moseq_train.PCAFit -->
+<g id="edge7" class="edge">
+<title>moseq_train.PCAPrep-&gt;moseq_train.PCAFit</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M276.75,-194.14C276.75,-182.84 276.75,-168.14 276.75,-158.06"/>
 </g>
 <!-- moseq_train.PreFitTask&#45;&gt;moseq_train.PreFit -->
-<g id="edge14" class="edge">
+<g id="edge8" class="edge">
 <title>moseq_train.PreFitTask-&gt;moseq_train.PreFit</title>
-<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M480.84,-61.66C480.84,-50.36 480.84,-35.66 480.84,-25.58"/>
-</g>
-<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.GridMoviesSampledInstances -->
-<g id="edge15" class="edge">
-<title>moseq_infer.Inference-&gt;moseq_infer.Inference.GridMoviesSampledInstances</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M655.23,-343.12C642.22,-330 617.69,-305.27 604.6,-292.07"/>
-</g>
-<!-- moseq_infer.Inference&#45;&gt;moseq_infer.Inference.MotionSequence -->
-<g id="edge16" class="edge">
-<title>moseq_infer.Inference-&gt;moseq_infer.Inference.MotionSequence</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M675.32,-347.09C702.88,-335.77 773.16,-306.92 809.04,-292.19"/>
-</g>
-<!-- moseq_infer.VideoRecording -->
-<g id="node19" class="node">
-<title>moseq_infer.VideoRecording</title>
-<g id="a_node19"><a xlink:title="→ session.Session
recording_id         
------------------------------
→ lab.Device
">
-<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="926.09,-511.2 755.59,-511.2 755.59,-476.64 926.09,-476.64 926.09,-511.2"/>
-<text text-anchor="start" x="763.59" y="-490.65" font-family="arial" text-decoration="underline" font-size="12.00" fill="darkgreen">moseq_infer.VideoRecording</text>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M480.75,-61.66C480.75,-50.36 480.75,-35.66 480.75,-25.58"/>
+</g>
+<!-- moseq_train.PCATask -->
+<g id="node10" class="node">
+<title>moseq_train.PCATask</title>
+<g id="a_node10"><a xlink:title="→ moseq_train.Bodyparts
------------------------------
kpms_project_output_dir=&quot;&quot; 
task_mode=&quot;load&quot;     
">
+<polygon fill="#00ff00" fill-opacity="0.188235" stroke="#00ff00" stroke-opacity="0.188235" points="343.62,-299.52 209.88,-299.52 209.88,-264.96 343.62,-264.96 343.62,-299.52"/>
+<text text-anchor="middle" x="276.75" y="-277.97" font-family="arial" font-size="12.00" fill="darkgreen">moseq_train.PCATask</text>
 </a>
 </g>
 </g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.VideoRecording.File -->
-<g id="edge17" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.VideoRecording.File</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M840.84,-476.32C840.84,-462.69 840.84,-443.92 840.84,-433.07"/>
+<!-- moseq_train.PCATask&#45;&gt;moseq_train.PCAPrep -->
+<g id="edge9" class="edge">
+<title>moseq_train.PCATask-&gt;moseq_train.PCAPrep</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M276.75,-264.64C276.75,-253.86 276.75,-239.87 276.75,-229.12"/>
 </g>
-<!-- moseq_infer.VideoRecording&#45;&gt;moseq_infer.InferenceTask -->
-<g id="edge18" class="edge">
-<title>moseq_infer.VideoRecording-&gt;moseq_infer.InferenceTask</title>
-<path fill="none" stroke="#000000" stroke-width="0.75" stroke-opacity="0.250980" d="M797.54,-476.15C770.01,-465.48 734.5,-451.73 707,-441.08"/>
+<!-- moseq_train.Bodyparts&#45;&gt;moseq_train.PCATask -->
+<g id="edge10" class="edge">
+<title>moseq_train.Bodyparts-&gt;moseq_train.PCATask</title>
+<path fill="none" stroke="#000000" stroke-width="2" stroke-opacity="0.250980" d="M276.75,-335.2C276.75,-324.42 276.75,-310.43 276.75,-299.68"/>
 </g>
 </g>
 </svg>
\ No newline at end of file