• Nathan Lambert's avatar
    Add UNet 1d for RL model for planning + colab (#105) · 7c5fef81
    Nathan Lambert authored
    * re-add RL model code
    
    * match model forward api
    
    * add register_to_config, pass training tests
    
    * fix tests, update forward outputs
    
    * remove unused code, some comments
    
    * add to docs
    
    * remove extra embedding code
    
    * unify time embedding
    
    * remove conv1d output sequential
    
    * remove sequential from conv1dblock
    
    * style and deleting duplicated code
    
    * clean files
    
    * remove unused variables
    
    * clean variables
    
    * add 1d resnet block structure for downsample
    
    * rename as unet1d
    
    * fix renaming
    
    * rename files
    
    * add get_block(...) api
    
    * unify args for model1d like model2d
    
    * minor cleaning
    
    * fix docs
    
    * improve 1d resnet blocks
    
    * fix tests, remove permuts
    
    * fix style
    
    * add output activation
    
    * rename flax blocks file
    
    * Add Value Function and corresponding example script to Diffuser implementation (#884)
    
    * valuefunction code
    
    * start example scripts
    
    * missing imports
    
    * bug fixes and pla...
    7c5fef81
README.md 287 Bytes