• Mufei Li's avatar
    [Model] Fix + batched DGMG (#175) · a0d0b1ea
    Mufei Li authored
    * DGMG with batch size 1
    
    * Fix
    
    * Adjustment
    
    * Fix
    
    * Fix
    
    * Fix
    
    * Fix
    
    * Fix has_node and __contains__
    
    * Batched implementation for DGMG
    
    * Remove redundant dependency
    
    * Adjustment
    
    * Fix
    
    * Add comments
    a0d0b1ea
main_batch.py 4.25 KB