Skip to content

Instantly share code, notes, and snippets.

@aidiary
Last active Nov 27, 2019
Embed
What would you like to do?
[PyTorch] Layerのタイプで処理を分岐
for layer in net.children():
if isinstance(layer, nn.Conv2d):
do something with the layer
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment