mirror of
https://github.com/gryf/coach.git
synced 2026-03-18 15:53:35 +01:00
removing datasets + imports optimization
This commit is contained in:
@@ -17,10 +17,10 @@
|
||||
from typing import List, Tuple, Union, Dict, Any
|
||||
|
||||
import numpy as np
|
||||
from rl_coach.utils import ReaderWriterLock
|
||||
|
||||
from rl_coach.core_types import Transition, Episode
|
||||
from rl_coach.memories.memory import Memory, MemoryGranularity, MemoryParameters
|
||||
from rl_coach.utils import ReaderWriterLock
|
||||
|
||||
|
||||
class EpisodicExperienceReplayParameters(MemoryParameters):
|
||||
|
||||
@@ -21,7 +21,8 @@ from typing import Tuple, List
|
||||
import numpy as np
|
||||
|
||||
from rl_coach.core_types import Episode, Transition
|
||||
from rl_coach.memories.episodic.episodic_experience_replay import EpisodicExperienceReplayParameters, EpisodicExperienceReplay
|
||||
from rl_coach.memories.episodic.episodic_experience_replay import EpisodicExperienceReplayParameters, \
|
||||
EpisodicExperienceReplay
|
||||
from rl_coach.memories.non_episodic.experience_replay import MemoryGranularity
|
||||
from rl_coach.spaces import GoalsSpace
|
||||
|
||||
|
||||
@@ -14,9 +14,8 @@
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
from rl_coach.memories.memory import MemoryGranularity, MemoryParameters
|
||||
|
||||
from rl_coach.memories.episodic.episodic_experience_replay import EpisodicExperienceReplay
|
||||
from rl_coach.memories.memory import MemoryGranularity, MemoryParameters
|
||||
|
||||
|
||||
class SingleEpisodeBufferParameters(MemoryParameters):
|
||||
|
||||
@@ -17,10 +17,10 @@
|
||||
from typing import List, Tuple, Union, Dict, Any
|
||||
|
||||
import numpy as np
|
||||
from rl_coach.utils import ReaderWriterLock
|
||||
|
||||
from rl_coach.core_types import Transition
|
||||
from rl_coach.memories.memory import Memory, MemoryGranularity, MemoryParameters
|
||||
from rl_coach.utils import ReaderWriterLock
|
||||
|
||||
|
||||
class ExperienceReplayParameters(MemoryParameters):
|
||||
|
||||
@@ -20,11 +20,11 @@ from enum import Enum
|
||||
from typing import List, Tuple, Any
|
||||
|
||||
import numpy as np
|
||||
from rl_coach.memories.memory import MemoryGranularity
|
||||
from rl_coach.schedules import Schedule, ConstantSchedule
|
||||
|
||||
from rl_coach.core_types import Transition
|
||||
from rl_coach.memories.memory import MemoryGranularity
|
||||
from rl_coach.memories.non_episodic.experience_replay import ExperienceReplayParameters, ExperienceReplay
|
||||
from rl_coach.schedules import Schedule, ConstantSchedule
|
||||
|
||||
|
||||
class PrioritizedExperienceReplayParameters(ExperienceReplayParameters):
|
||||
|
||||
Reference in New Issue
Block a user