github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

[chore] Update placement_policy.py

pull/5722/head
botbw 6 months ago committed by GitHub
parent
013690a86b
commit
e57812c672
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 5 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 5
      colossalai/zero/gemini/placement_policy.py

5
colossalai/zero/gemini/placement_policy.py
Unescape View File

@ -45,11 +45,6 @@ class PlacementPolicy(ABC):
raise NotImplementedError raise NotImplementedError
import os
rank = int(os.environ["RANK"])
class StaticPlacementPolicy(PlacementPolicy): class StaticPlacementPolicy(PlacementPolicy):
def __init__( def __init__(
self, self,

Write Preview
Loading…
Cancel
Save