n0obcoder

n0obcoder / understanding_pytorch___00___import_libraries.py

Last active August 12, 2019 18:20

	import torch
	import torch.nn as nn

n0obcoder / understanding_pytorch___02___print_model.py

Last active August 12, 2019 18:19

n0obcoder / understanding_pytorch___03___named_parameters.py

Last active August 12, 2019 18:19

	for name, param in model.named_parameters():
	print('name: ', name)
	print(type(param))
	print('param.shape: ', param.shape)
	print('param.requires_grad: ', param.requires_grad)
	print('=====')

n0obcoder / understanding_pytorch___04___freeze_all_but_fc.py

Last active August 13, 2019 06:51

	for name, param in model.named_parameters():
	if name in ['fc.weight', 'fc.bias']:
	param.requires_grad = True
	else:
	param.requires_grad = False

n0obcoder / understanding_pytorch___05___verifying_freeze.py

Last active August 12, 2019 18:18

	for name, param in model.named_parameters():
	print(name, ':', param.requires_grad)

n0obcoder / understanding_pytorch___06___state_dict.py

Last active August 12, 2019 18:18

model.state_dict()

n0obcoder / understanding_pytorch___07___help_model.py

Last active August 12, 2019 18:18

help(model)

n0obcoder / understanding_pytorch___08___1_isinstance_model.py

Last active August 12, 2019 18:17

isinstance(model, nn.Module)

n0obcoder / understanding_pytorch___08___2_isinstance_model_fc.py

Last active August 12, 2019 18:16

isinstance(model.fc, nn.Module)

n0obcoder / understanding_pytorch___09___named_children.py

Last active August 12, 2019 18:16

	for name, child in model.named_children():
	print('name: ', name)
	print('isinstance({}, nn.Module): '.format(name), isinstance(child, nn.Module))
	print('=====')