Add files using upload-large-folder tool

aa30701 verified over 1 year ago

330 Bytes

import torch
import yaml



# conda create -n openai ipykernel
def get_model_num_params(model):
    # return sum(p.numel() for p in model.parameters() if p.requires_grad)
    return sum(p.numel() for p in model.parameters())


def load_yaml(path):
    with open(path, 'r') as f:
        return yaml.load(f, Loader=yaml.FullLoader)

	import torch
	import yaml



	# conda create -n openai ipykernel
	def get_model_num_params(model):
	# return sum(p.numel() for p in model.parameters() if p.requires_grad)
	return sum(p.numel() for p in model.parameters())


	def load_yaml(path):
	with open(path, 'r') as f:
	return yaml.load(f, Loader=yaml.FullLoader)