All Questions
Tagged with stablebaseline3 pytorch
1 question
2
votes
1
answer
235
views
Training a Custom Feature Extractor in Stable Baselines3 Starting from Pre-trained Weights?
I am using the following custom feature extractor for my StableBaselines3 model:
import torch.nn as nn
from stable_baselines3 import PPO
class Encoder(nn.Module):
def __init__(self, input_dim, ...