We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aadaca9 commit a303586Copy full SHA for a303586
ml-agents/mlagents/trainers/models_torch.py
@@ -507,7 +507,6 @@ def get_encoder_for_type(encoder_type: EncoderType) -> nn.Module:
507
EncoderType.NATURE_CNN: NatureVisualEncoder,
508
EncoderType.RESNET: ResNetVisualEncoder,
509
}
510
- print(encoder_type, ENCODER_FUNCTION_BY_TYPE.get(encoder_type))
511
return ENCODER_FUNCTION_BY_TYPE.get(encoder_type)
512
513
@staticmethod
0 commit comments