ejmejm

from typing import Tuple

import equinox as eqx
import jax
from jax import Array
import jax.numpy as jnp

from cl_agent.utils import tree_replace

ejmejm

device: cuda

wandb:
  enabled: false
  project: minigrid_gvfs

optimizer:
  type: sgd
  meta_lr: 0.01
  #...

ejmejm

import math

import torch
from torch.optim.optimizer import Optimizer
from typing import Iterator, Optional


class IDBD(Optimizer):
    """Incremental Delta-Bar-Delta optimizer.
    

ejmejm

from equinox import nn
import jax
import jax.numpy as jnp


#########################################
#               Test 1                  #
#########################################

ejmejm

ejmejm