Row 95930

Row ID: 95930 | Dataset Entry | Axioma AXP Content Repository

Content Data

This page contains data entry 95930 from the Axioma AXP content repository. The structured data below represents the complete record for this entry.

I have an RL Python code that I want to speed up with JIT.

I have changed from the class definition (torch.nn.Module) to (torch.jit.ScriptModule) and added the decorator u/torch.jit.script\_method. I need to rerun the numbers, but my impression is that it speeds up slightly the training.

If I print the layers I can see: (conv2\_q1): RecursiveScriptModule(original\_name=Conv2d)

What else can I speed up with JIT? Can I set up the training part with JIT?

Also, how does this all tie with torch.jit.trace and torch.jit.script?

It is a beginner question, I am quite new to this possible optimization. Feel free to refer to any training material to understand everything.

Thanks!

FieldValue
text I have an RL Python code that I want to speed up with JIT. I have changed from the class definition (torch.nn.Module) to (torch.jit.ScriptModule) and added the decorator u/torch.jit.script\_method. I need to rerun the numbers, but my impression is that it speeds up slightly the training. If I print the layers I can see: (conv2\_q1): RecursiveScriptModule(original\_name=Conv2d) What else can I speed up with JIT? Can I set up the training part with JIT? Also, how does this all tie with torch.j…
label r/pytorch
dataType post
communityName r/pytorch
datetime 2024-05-25
username_encoded Z0FBQUFBQm5Lak11alBRU2FsV2l0bDU4S2k2aGNkb0RvdEdGM3VzeWgwTHI1RTAycEhCV3F4SFNYSTN4VzVGRGZpa29DbWE5R01VMFlkaHNNYTVxbU1qNURXOVZ4NFQxdFk0UDdJMkdJc2wyc001a0p6bEZzOFE9
url_encoded Z0FBQUFBQm5LalBBRWUzVllUdUFUZFZrNFZtUFZ5UFBJc0FQUlZrTjNULThyVlZnVF9USVZlbE9IQ2RCUVVpTnh6SDNtbHRIeml1WnhRa3A3UDZNYjNXTFlsY3ZFeGV1YzEzYVNOczB6UmZmWURLRXFpYXNuMlEtd3k5QUtXTFVNLWE5Q1l3TlBrQ2JRLUtadldiMHdZU0ZrTVRJVGVDeWhFNG5OUmc0VDh4Z1lCWlpJTzN2MDNnPQ==

Raw Record

{
  "text": "I have an RL Python code that I want to speed up with JIT.\n\nI have changed from the class definition (torch.nn.Module) to (torch.jit.ScriptModule) and added the decorator u/torch.jit.script\\_method. I need to rerun the numbers, but my impression is that it speeds up slightly the training.\n\nIf I print the layers I can see: (conv2\\_q1): RecursiveScriptModule(original\\_name=Conv2d)\n\nWhat else can I speed up with JIT? Can I set up the training part with JIT?\n\nAlso, how does this all tie with torch.jit.trace and torch.jit.script?\n\nIt is a beginner question, I am quite new to this possible optimization. Feel free to refer to any training material to understand everything.\n\nThanks!",
  "label": "r/pytorch",
  "dataType": "post",
  "communityName": "r/pytorch",
  "datetime": "2024-05-25",
  "username_encoded": "Z0FBQUFBQm5Lak11alBRU2FsV2l0bDU4S2k2aGNkb0RvdEdGM3VzeWgwTHI1RTAycEhCV3F4SFNYSTN4VzVGRGZpa29DbWE5R01VMFlkaHNNYTVxbU1qNURXOVZ4NFQxdFk0UDdJMkdJc2wyc001a0p6bEZzOFE9",
  "url_encoded": "Z0FBQUFBQm5LalBBRWUzVllUdUFUZFZrNFZtUFZ5UFBJc0FQUlZrTjNULThyVlZnVF9USVZlbE9IQ2RCUVVpTnh6SDNtbHRIeml1WnhRa3A3UDZNYjNXTFlsY3ZFeGV1YzEzYVNOczB6UmZmWURLRXFpYXNuMlEtd3k5QUtXTFVNLWE5Q1l3TlBrQ2JRLUtadldiMHdZU0ZrTVRJVGVDeWhFNG5OUmc0VDh4Z1lCWlpJTzN2MDNnPQ=="
}
Explore Dataset Explore Row

Entry Information