Remove warnings (#111)

- Replace `use_pretrained_backbone` with `pretrained_backbone_weights`
- Bump diffusers' minimum version `0.26.3` -> `0.27.2`
- Add ignore flags in CI's pytest
- Change Box observation spaces in simulation environments
- Set `version_base="1.2"` in Hydra initializations
- Bump einops' minimum version `0.7.0` -> `0.8.0`
This commit is contained in:
Simon Alibert
2024-04-29 00:31:33 +02:00
committed by GitHub
parent 55dc9f7f51
commit 791506dfb8
14 changed files with 52 additions and 36 deletions

View File

@@ -45,7 +45,7 @@ policy:
# Architecture.
# Vision backbone.
vision_backbone: resnet18
use_pretrained_backbone: true
pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
replace_final_stride_with_dilation: false
# Transformer layers.
pre_norm: false

View File

@@ -61,7 +61,7 @@ policy:
vision_backbone: resnet18
crop_shape: [84, 84]
crop_is_random: True
use_pretrained_backbone: false
pretrained_backbone_weights: null
use_group_norm: True
spatial_softmax_num_keypoints: 32
# Unet.