ObsEncoder

class torchrl.modules.ObsEncoder(channels=32, num_layers=4, depth=None)

觀察編碼器網路。

取得像素觀察並將其編碼為潛在空間。

參考： https://arxiv.org/abs/1803.10122

參數:

• channels (int, optional) – 第一層中的隱藏單元數。預設為 32。

• num_layers (int, optional) – 網路深度。預設為 4。

forward(observation)

定義每次呼叫時執行的計算。

應由所有子類別覆寫。

注意

雖然 forward pass 的配方需要在這個函式中定義，但應該在之後呼叫 Module 實例，而不是這個函式，因為前者會處理執行已註冊的 hooks，而後者會靜默地忽略它們。