Skip to main content

All Questions

Filter by
Sorted by
Tagged with
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, ...
Sayyor Y's user avatar
  • 1,286