Skip to content

Instantly share code, notes, and snippets.

View alexlimh's full-sized avatar

Minghan Li alexlimh

View GitHub Profile
@alexlimh
alexlimh / dreamer.py
Last active April 29, 2020 14:02
dreamer.py: add config.reuse_actor; add EmpowStepActionBehavior models.py: add EmpowStepActionBehavior; change EmpowVIMBehavior; networks.py: change ActionHead class.
import argparse
import collections
import functools
import gc
import os
import pathlib
import resource
import sys
import warnings
@alexlimh
alexlimh / dreamer.py
Created April 29, 2020 13:58
dreamer.py
import argparse
import collections
import functools
import gc
import os
import pathlib
import resource
import sys
import warnings
ATARI_DEFAULTS = dict(
# General
steps=2e8,
eval_every=1e5,
log_every=1e4,
prefill=5e4,
dataset_size=4e6,
pretrain=0,
# Environment
import argparse
import collections
import functools
import gc
import os
import pathlib
import resource
import sys
import warnings